You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
864 B
23 lines
864 B
from .base_options import BaseOptions
|
|
|
|
|
|
class TrainOptions(BaseOptions):
|
|
"""This class includes training options.
|
|
|
|
It also includes shared options defined in BaseOptions.
|
|
"""
|
|
|
|
def initialize(self, parser):
|
|
parser = BaseOptions.initialize(self, parser)
|
|
# visdom and HTML visualization parameters
|
|
|
|
# network saving and loading parameters
|
|
parser.add_argument('--phase', type=str, default='train', help='train, val, test, etc')
|
|
|
|
# training parameters
|
|
parser.add_argument('--epochs', type=int, default=10000, help='number of epochs')
|
|
parser.add_argument('--lr', type=float, default=0.001, help='initial learning rate for adam')
|
|
parser.add_argument('--mod', type=str, default='mod2', help='chooses which dataset model for train. mod1....')
|
|
|
|
self.isTrain = True
|
|
return parser
|