<filename>options/train_opts.py
from .base_opts import BaseOpts
class TrainOpts(BaseOpts):
    def __init__(self):
        super(TrainOpts, self).__init__()
        self.initialize()

    def initialize(self):
        BaseOpts.initialize(self)
        #### Dataset Arguments ####
        self.parser.add_argument('--dataset',     default='PS_Synth_Dataset')
        self.parser.add_argument('--data_dir',    default='data/datasets/PS_Blobby_Dataset')
        self.parser.add_argument('--data_dir2',   default='data/datasets/PS_Sculpture_Dataset')
        self.parser.add_argument('--concat_data', default=False, action='store_true')
        self.parser.add_argument('--rescale',     default=True,  action='store_false')
        self.parser.add_argument('--crop',        default=True,  action='store_false')
        self.parser.add_argument('--crop_h',      default=32,    type=int)
        self.parser.add_argument('--crop_w',      default=32,    type=int)
        self.parser.add_argument('--noise_aug',   default=True,  action='store_false')
        self.parser.add_argument('--noise',       default=0.05,  type=float)
        self.parser.add_argument('--color_aug',   default=True,  action='store_false') 

        #### Training Arguments ####
        self.parser.add_argument('--model',       default='PS_FCN')
        self.parser.add_argument('--solver',      default='adam', help='adam|sgd')
        self.parser.add_argument('--milestones',  default=[5, 10, 15, 20, 25], nargs='+', type=int)
        self.parser.add_argument('--init_lr',     default=1e-3, type=float)
        self.parser.add_argument('--lr_decay',    default=0.5,  type=float)
        self.parser.add_argument('--beta_1',      default=0.9,  type=float, help='adam')
        self.parser.add_argument('--beta_2',      default=0.999,type=float, help='adam')
        self.parser.add_argument('--momentum',    default=0.9,  type=float, help='sgd')
        self.parser.add_argument('--batch',       default=32,   type=int)
        self.parser.add_argument('--val_batch',   default=8,    type=int)

        #### Display Arguments ####
        self.parser.add_argument('--train_disp',  default=20,   type=int)
        self.parser.add_argument('--train_save',  default=200,  type=int)
        self.parser.add_argument('--val_intv',    default=1,    type=int)
        self.parser.add_argument('--val_disp',    default=1,    type=int)
        self.parser.add_argument('--val_save',    default=1,    type=int)

        #### Checkpoint Arguments ####
        self.parser.add_argument('--save_intv',  default=1, type=int)

        #### Loss Arguments ####
        self.parser.add_argument('--normal_loss',default='cos', help='cos|mse')
        self.parser.add_argument('--normal_w',   default=1)

    def parse(self):
        BaseOpts.parse(self)
        return self.args

<filename>options/train_opts.py
from .base_opts import BaseOpts
class TrainOpts(BaseOpts):
    def __init__(self):
        super(TrainOpts, self).__init__()
        self.initialize()

    def initialize(self):
        BaseOpts.initialize(self)
        #### Dataset Arguments ####
        self.parser.add_argument('--dataset',     default='PS_Synth_Dataset')
        self.parser.add_argument('--data_dir',    default='data/datasets/PS_Blobby_Dataset')
        self.parser.add_argument('--data_dir2',   default='data/datasets/PS_Sculpture_Dataset')
        self.parser.add_argument('--concat_data', default=False, action='store_true')
        self.parser.add_argument('--rescale',     default=True,  action='store_false')
        self.parser.add_argument('--crop',        default=True,  action='store_false')
        self.parser.add_argument('--crop_h',      default=32,    type=int)
        self.parser.add_argument('--crop_w',      default=32,    type=int)
        self.parser.add_argument('--noise_aug',   default=True,  action='store_false')
        self.parser.add_argument('--noise',       default=0.05,  type=float)
        self.parser.add_argument('--color_aug',   default=True,  action='store_false') 

        #### Training Arguments ####
        self.parser.add_argument('--model',       default='PS_FCN')
        self.parser.add_argument('--solver',      default='adam', help='adam|sgd')
        self.parser.add_argument('--milestones',  default=[5, 10, 15, 20, 25], nargs='+', type=int)
        self.parser.add_argument('--init_lr',     default=1e-3, type=float)
        self.parser.add_argument('--lr_decay',    default=0.5,  type=float)
        self.parser.add_argument('--beta_1',      default=0.9,  type=float, help='adam')
        self.parser.add_argument('--beta_2',      default=0.999,type=float, help='adam')
        self.parser.add_argument('--momentum',    default=0.9,  type=float, help='sgd')
        self.parser.add_argument('--batch',       default=32,   type=int)
        self.parser.add_argument('--val_batch',   default=8,    type=int)

        #### Display Arguments ####
        self.parser.add_argument('--train_disp',  default=20,   type=int)
        self.parser.add_argument('--train_save',  default=200,  type=int)
        self.parser.add_argument('--val_intv',    default=1,    type=int)
        self.parser.add_argument('--val_disp',    default=1,    type=int)
        self.parser.add_argument('--val_save',    default=1,    type=int)

        #### Checkpoint Arguments ####
        self.parser.add_argument('--save_intv',  default=1, type=int)

        #### Loss Arguments ####
        self.parser.add_argument('--normal_loss',default='cos', help='cos|mse')
        self.parser.add_argument('--normal_w',   default=1)

    def parse(self):
        BaseOpts.parse(self)
        return self.args

#### Dataset Arguments #### #### Training Arguments #### #### Display Arguments #### #### Checkpoint Arguments #### #### Loss Arguments ####