File size: 317 Bytes
8ec10cf |
1 2 3 4 5 6 7 8 9 10 |
def set_template(args):
if args.template.find('gopro') >= 0:
args.dataset = 'GOPRO_Large'
args.milestones = [500, 750, 900]
args.end_epoch = 1000
elif args.template.find('reds') >= 0:
args.dataset = 'REDS'
args.milestones = [100, 150, 180]
args.end_epoch = 200
|