diff --git a/image_ref/config.py b/image_ref/config.py index 15f55134d9bbc066f346d2c0c551ec790a4abd76..7137615b664c287ce52958539a0c2dd691ff5e45 100644 --- a/image_ref/config.py +++ b/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('-- dataset_ref_dir', 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)