training main.py boilerplate

main.py

import argparse

def str2bool(v):
    if isinstance(v, bool):
        return v
    if v.lower() in ('yes', 'true', 't', 'y', '1'):
        return True
    elif v.lower() in ('no', 'false', 'f', 'n', '0'):
        return False
    else:
        raise argparse.ArgumentTypeError('Boolean value expected.')

parser = argparse.ArgumentParser()
parser.add_argument('--expname', type=str, default=None, help='experiment name of wandb')
parser.add_argument('--batch_size', type=int, default=32)
parser.add_argument('--save_model', type=str2bool, nargs='?', const=True, default=False)
args = parser.parse_args()

def main():
    train(args)
    
if __name__ == '__main__':
    main()