diff --git a/main_ray.py b/main_ray.py index a79fb47fc4449d16a87e05a4a2a4464c3ca4fa3a..cbad14268d1627e5f1989fc975b62068a0de82dc 100644 --- a/main_ray.py +++ b/main_ray.py @@ -265,8 +265,7 @@ def analyze_ray_res(storage_path, exp_name, args, gpus_per_trial=1): 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 cont acc of :", - result.metrics["val cont acc"], " val acc of :", + result.metrics["val loss"], " val acc of :", result.metrics["val acc"] )