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