diff --git a/main_ray.py b/main_ray.py index 6191d03881cfda43c6fbd657967d4daf5a71d5a0..e2bf1a0666e22c0537ee4bc89272284dbe59d425 100644 --- a/main_ray.py +++ b/main_ray.py @@ -256,12 +256,16 @@ def analyze_ray_res(storage_path, exp_name, args, gpus_per_trial=1): num_results = len(results) print("Number of results:", num_results) + df = results.get_dataframe() + df.to_csv('df_results_calmettes.csv') + for i, result in enumerate(results): if result.error: print(f"Trial #{i} had an error:", result.error) continue - print( + else : + print( f"Trial #{i} finished successfully train loss of :", {result.metrics["train loss"]}, " train acc of :", {result.metrics["train cont acc"]}, " val loss of :", @@ -269,8 +273,7 @@ def analyze_ray_res(storage_path, exp_name, args, gpus_per_trial=1): {result.metrics["val acc"]} ) - df = results.get_dataframe() - df.to_csv('df_results_calmettes.csv') +