diff --git a/barlow_twin_like/config.py b/barlow_twin_like/config.py index ee4a58b211b7b5a047f62cf8d7ab87cd8acef510..a9aeb92a95bc0c883ca972f72bfdd30d1b30eaf8 100644 --- a/barlow_twin_like/config.py +++ b/barlow_twin_like/config.py @@ -13,7 +13,7 @@ def load_args_barlow(): parser.add_argument('--lambd', type=float, default=0.005) parser.add_argument('--opti', type=str, default='adam') parser.add_argument('--model', type=str, default='ResNet18') - parser.add_argument('--projector', type=str, default='256-128-64') + parser.add_argument('--projector', type=str, default='2048-4096') parser.add_argument('--sampler', type=str, default=None) parser.add_argument('--dataset_train_dir', type=str, default='data/processed_data_wiff_clean_005_10000_apex/npy_image/train data') parser.add_argument('--dataset_val_dir', type=str, default='data/processed_data_wiff_clean_005_10000_apex/npy_image/val data')