def set_template(args):
    # Set the templates here
    if args.template.find('bisrnet') >= 0:
        args.input_setting = 'H'
        args.input_mask = 'Mask'
        args.scheduler = 'CosineAnnealingLR'