Skip to content
Snippets Groups Projects
Commit 4ad03260 authored by Schneider Leo's avatar Schneider Leo
Browse files

fix : ref_count_thr args ray contrastive

parent c73ed9a3
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 2 additions and 2 deletions
image_ref/main_ray.py
+ 2
− 2
View file @ 4ad03260
...@@ -19,9 +19,9 @@ from ray.tune.schedulers import ASHAScheduler ...@@ -19,9 +19,9 @@ from ray.tune.schedulers import ASHAScheduler
def train_model(config,args): def train_model(config,args):
# load data # 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' 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' ref_dir = '/lustre/fswork/projects/rech/bun/ucg81ws/these/pseudo_image/image_ref/img_ref_count_th_10'
else : else :
ref_dir = '/lustre/fswork/projects/rech/bun/ucg81ws/these/pseudo_image/image_ref/img_ref_count_th_5' ref_dir = '/lustre/fswork/projects/rech/bun/ucg81ws/these/pseudo_image/image_ref/img_ref_count_th_5'
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment