diff --git a/image_ref/main_ray.py b/image_ref/main_ray.py index 0d678aa7d1a4a34c85951d42fd679aa6242fce06..80fa177ad708f7c00a2a7eb16d18ff5135c50164 100644 --- a/image_ref/main_ray.py +++ b/image_ref/main_ray.py @@ -294,6 +294,8 @@ 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_contrastive.csv') for i, result in enumerate(results): if result.error: print(f"Trial #{i} had an error:", result.error) @@ -311,6 +313,6 @@ def analyze_ray_res(storage_path, exp_name,args, gpus_per_trial=1): if __name__ == '__main__': args = load_args_contrastive() - analyze_ray_res() + analyze_ray_res('/lustre/fswork/projects/rech/bun/ucg81ws/these/pseudo_image/image_ref/ray_results/','ref_count_threshold_experiment',args) # print(args) # main(args)