| 20 | } |
| 21 | |
| 22 | def initialize(self, parser): |
| 23 | # specify folder |
| 24 | parser.add_argument('--data_folder', type=str, default='./data', |
| 25 | help='path to data') |
| 26 | parser.add_argument('--model_path', type=str, default='./save', |
| 27 | help='path to save model') |
| 28 | parser.add_argument('--tb_path', type=str, default='./tb', |
| 29 | help='path to tensorboard') |
| 30 | |
| 31 | # basics |
| 32 | parser.add_argument('--print_freq', type=int, default=10, |
| 33 | help='print frequency') |
| 34 | parser.add_argument('--save_freq', type=int, default=20, |
| 35 | help='save frequency') |
| 36 | parser.add_argument('--batch_size', type=int, default=256, |
| 37 | help='batch_size') |
| 38 | parser.add_argument('-j', '--num_workers', type=int, default=40, |
| 39 | help='num of workers to use') |
| 40 | |
| 41 | # optimization |
| 42 | parser.add_argument('--epochs', type=int, default=200, |
| 43 | help='number of training epochs') |
| 44 | parser.add_argument('--learning_rate', type=float, default=0.03, |
| 45 | help='learning rate') |
| 46 | parser.add_argument('--lr_decay_epochs', type=str, default='120,160', |
| 47 | help='where to decay lr, can be a list') |
| 48 | parser.add_argument('--lr_decay_rate', type=float, default=0.1, |
| 49 | help='decay rate for learning rate') |
| 50 | parser.add_argument('--weight_decay', type=float, default=1e-4, |
| 51 | help='weight decay') |
| 52 | parser.add_argument('--momentum', type=float, default=0.9, |
| 53 | help='momentum for SGD') |
| 54 | parser.add_argument('--cosine', action='store_true', |
| 55 | help='using cosine annealing') |
| 56 | |
| 57 | # method selection |
| 58 | parser.add_argument('--method', default='Customize', type=str, |
| 59 | choices=['InsDis', 'CMC', 'CMCv2', 'MoCo', 'MoCov2', |
| 60 | 'PIRL', 'InfoMin', 'Customize'], |
| 61 | help='Choose predefined method. Configs will be override ' |
| 62 | 'for all methods except for `Customize`, which allows ' |
| 63 | 'for user-defined combination of methods') |
| 64 | # method configuration |
| 65 | parser.add_argument('--modal', default='RGB', type=str, choices=['RGB', 'CMC'], |
| 66 | help='single RGB modal, or two modalities in CMC') |
| 67 | parser.add_argument('--jigsaw', action='store_true', |
| 68 | help='adding PIRL branch') |
| 69 | parser.add_argument('--mem', default='bank', type=str, choices=['bank', 'moco'], |
| 70 | help='memory mechanism: memory bank, or moco encoder cache') |
| 71 | |
| 72 | # model setup |
| 73 | parser.add_argument('--arch', default='resnet50', type=str, |
| 74 | help='e.g., resnet50, resnext50, resnext101' |
| 75 | 'and their wider variants, resnet50x4') |
| 76 | parser.add_argument('-d', '--feat_dim', default=128, type=int, |
| 77 | help='feature dimension for contrastive loss') |
| 78 | parser.add_argument('-k', '--nce_k', default=65536, type=int, |
| 79 | help='number of negatives') |