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

fix : file config

parent 9c0725e6
No related branches found
No related tags found
No related merge requests found
...@@ -10,6 +10,7 @@ def load_args_contrastive(): ...@@ -10,6 +10,7 @@ def load_args_contrastive():
parser.add_argument('--noise_threshold', type=int, default=0) parser.add_argument('--noise_threshold', type=int, default=0)
parser.add_argument('--lr', type=float, default=0.001) parser.add_argument('--lr', type=float, default=0.001)
parser.add_argument('--batch_size', type=int, default=64) parser.add_argument('--batch_size', type=int, default=64)
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')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment