From 4ad03260e6d5ceee5f2f35dfa5835a1b03c39ea3 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Fri, 18 Apr 2025 08:31:47 +0200
Subject: [PATCH] fix : ref_count_thr args ray contrastive
---
image_ref/main_ray.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/image_ref/main_ray.py b/image_ref/main_ray.py
index b06aa9b..d0ec7c0 100644
--- a/image_ref/main_ray.py
+++ b/image_ref/main_ray.py
@@ -19,9 +19,9 @@ from ray.tune.schedulers import ASHAScheduler
def train_model(config,args):
# load data
- if config['res_count_thr']=='none':
+ if config['ref_count_thr']=='none':
ref_dir = '/lustre/fswork/projects/rech/bun/ucg81ws/these/pseudo_image/image_ref/img_ref'
- elif config['res_count_thr']=='10':
+ elif config['ref_count_thr']=='10':
ref_dir = '/lustre/fswork/projects/rech/bun/ucg81ws/these/pseudo_image/image_ref/img_ref_count_th_10'
else :
ref_dir = '/lustre/fswork/projects/rech/bun/ucg81ws/these/pseudo_image/image_ref/img_ref_count_th_5'
--
GitLab