diff --git a/main_ray.py b/main_ray.py index cbad14268d1627e5f1989fc975b62068a0de82dc..5c88881597841f0a29413b39c93f4dacada33205 100644 --- a/main_ray.py +++ b/main_ray.py @@ -262,13 +262,16 @@ def analyze_ray_res(storage_path, exp_name, args, gpus_per_trial=1): continue print( - f"Trial #{i} finished successfully train loss of :", - result.metrics["train loss"], " train acc of :", - result.metrics["train cont acc"], " val loss of :", - result.metrics["val loss"], " val acc of :", - result.metrics["val acc"] + f"Trial #{i} ({result.trial_id}) finished successfully train loss of :", + {result.metrics["train loss"]}, " train acc of :", + {result.metrics["train cont acc"]}, " val loss of :", + {result.metrics["val loss"]}, " val acc of :", + {result.metrics["val acc"]} ) + df = results.get_dataframe() + df.to_csv('df_results_calmettes.csv') + if __name__ == '__main__':