diff --git a/image_ref/config.py b/image_ref/config.py index 58a4c48bd9c1f0078ca99c06194db9c031dd9db4..26c1bdfe544fc78f6309a2908b77fe43ba54611e 100644 --- a/image_ref/config.py +++ b/image_ref/config.py @@ -13,7 +13,7 @@ def load_args_contrastive(): parser.add_argument('--positive_prop', type=int, default=None) parser.add_argument('--model', type=str, default='ResNet18') parser.add_argument('--model_type', type=str, default='duo') - parser.add_argument('--dataset_dir', type=str, default='data/processed_data/npy_image/data_training') + parser.add_argument('--dataset_dir', type=str, default='data/processed_data/npy_image/data_training_contrastive') 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')