fix : config

image_ref/config.py

@@ -17,7 +17,7 @@ def load_args_contrastive():
     parser.add_argument('--dataset_val_dir', type=str, default='data/processed_data/npy_image/data_test_contrastive')
     parser.add_argument('--dataset_test_dir', type=str, default=None)
     parser.add_argument('--base_out', type=str, default='output/baseline')
-    parser.add_argument('-- ', type=str, default='image_ref/img_ref')
+    parser.add_argument('-- dataset_ref_dir', type=str, default='image_ref/img_ref')
     parser.add_argument('--output', type=str, default='output/out_contrastive.csv')
     parser.add_argument('--save_path', type=str, default='output/best_model_constrastive.pt')
     parser.add_argument('--pretrain_path', type=str, default=None)