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

fix : file config

parent e36e38d7
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 1 addition and 1 deletion
image_ref/config.py
+ 1
− 1
View file @ dc91f876
...@@ -13,7 +13,7 @@ def load_args_contrastive(): ...@@ -13,7 +13,7 @@ def load_args_contrastive():
parser.add_argument('--positive_prop', type=int, default=None) parser.add_argument('--positive_prop', type=int, default=None)
parser.add_argument('--model', type=str, default='ResNet18') parser.add_argument('--model', type=str, default='ResNet18')
parser.add_argument('--model_type', type=str, default='duo') 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('--dataset_ref_dir', type=str, default='image_ref/img_ref')
parser.add_argument('--output', type=str, default='output/out_contrastive.csv') 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('--save_path', type=str, default='output/best_model_constrastive.pt')
......
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