Spaces:
Sleeping
Sleeping
File size: 804 Bytes
9c4b1c4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from .base_options import BaseOptions
class TestOptions(BaseOptions):
def initialize(self, parser):
parser = BaseOptions.initialize(self, parser)
# parser.add_argument('--dataroot')
parser.add_argument('--model_path')
parser.add_argument('--no_resize', action='store_true')
parser.add_argument('--no_crop', action='store_true')
parser.add_argument('--eval', action='store_true', help='use eval mode during test time.')
parser.add_argument('--earlystop_epoch', type=int, default=15)
parser.add_argument('--lr', type=float, default=0.00002, help='initial learning rate for adam')
parser.add_argument('--niter', type=int, default=0, help='# of iter at starting learning rate')
self.isTrain = False
return parser
|