diff --git a/image_ref/config.py b/image_ref/config.py index 3a28cf8b9e7d107c5c9cad7dff88a2f32ee8a47e..455bfa13fbd44b0552ecbd76debae6e6a428a87e 100644 --- a/image_ref/config.py +++ b/image_ref/config.py @@ -12,10 +12,10 @@ def load_args_contrastive(): parser.add_argument('--batch_size', type=int, default=64) 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_ref_dir', type=str, default='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('--dataset_dir', type=str, default='data/processed_data/npy_image/data_training') + 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) args = parser.parse_args()