Class BaseOptions

pycontrast/options/base_options.py:5–155

class BaseOptions(object):

    def __init__(self):
        self.initialized = False
        self.parser = None
        self.opt = None
        # config for predefined method
        self.override_dict = {
            'InsDis':  ['RGB', False, 'bank', 'A', 'linear', 0.07],
            'CMC':     ['CMC', False, 'bank', 'C', 'linear', 0.07],
            'MoCo':    ['RGB', False, 'moco', 'A', 'linear', 0.07],
            'PIRL':    ['RGB', True,  'bank', 'A', 'linear', 0.07],
            'MoCov2':  ['RGB', False, 'moco', 'B', 'mlp',    0.2],
            'CMCv2':   ['CMC', False, 'moco', 'E', 'mlp',    0.2],
            'InfoMin': ['RGB', True,  'moco', 'D', 'mlp',    0.15],
        }

    def initialize(self, parser):
        # specify folder
        parser.add_argument('--data_folder', type=str, default='./data',
                            help='path to data')
        parser.add_argument('--model_path', type=str, default='./save',
                            help='path to save model')
        parser.add_argument('--tb_path', type=str, default='./tb',
                            help='path to tensorboard')

        # basics
        parser.add_argument('--print_freq', type=int, default=10,
                            help='print frequency')
        parser.add_argument('--save_freq', type=int, default=20,
                            help='save frequency')
        parser.add_argument('--batch_size', type=int, default=256,
                            help='batch_size')
        parser.add_argument('-j', '--num_workers', type=int, default=40,
                            help='num of workers to use')

        # optimization
        parser.add_argument('--epochs', type=int, default=200,
                            help='number of training epochs')
        parser.add_argument('--learning_rate', type=float, default=0.03,
                            help='learning rate')
        parser.add_argument('--lr_decay_epochs', type=str, default='120,160',
                            help='where to decay lr, can be a list')
        parser.add_argument('--lr_decay_rate', type=float, default=0.1,
                            help='decay rate for learning rate')
        parser.add_argument('--weight_decay', type=float, default=1e-4,
                            help='weight decay')
        parser.add_argument('--momentum', type=float, default=0.9,
                            help='momentum for SGD')
        parser.add_argument('--cosine', action='store_true',
                            help='using cosine annealing')

        # method selection
        parser.add_argument('--method', default='Customize', type=str,
                            choices=['InsDis', 'CMC', 'CMCv2', 'MoCo', 'MoCov2',
                                     'PIRL', 'InfoMin', 'Customize'],
                            help='Choose predefined method. Configs will be override '
                                 'for all methods except for `Customize`, which allows '