def parse_args()

in main.py [0:0]

• 24 lines of code
• 1 McCabe index (conditional complexity)

def parse_args():
    parser = argparse.ArgumentParser(description='Main training script')
    parser.add_argument('--traincfg', required=True, help='yaml file containing config for data')
    parser.add_argument('--valcfg', required=True, help='yaml file containing config for data')
    parser.add_argument('--model', default='ResNet18', help='model: ResNet{10|18|34|50}')
    parser.add_argument('--lr', default=0.1, type=float, help='Initial learning rate')
    parser.add_argument('--momentum', default=0.9, type=float, help='Momentum')
    parser.add_argument('--weight_decay', default=0.0001, type=float, help='Weight decay')
    parser.add_argument('--lr_decay', default=0.1, type=float, help='Learning rate decay')
    parser.add_argument('--step_size', default=30, type=int, help='Step size')
    parser.add_argument('--print_freq', default=10, type=int,help='Print frequecy')
    parser.add_argument('--save_freq', default=10, type=int, help='Save frequency')
    parser.add_argument('--start_epoch', default=0, type=int,help ='Starting epoch')
    parser.add_argument('--stop_epoch', default=90, type=int, help ='Stopping epoch')
    parser.add_argument('--allow_resume', default=0, type=int)
    parser.add_argument('--resume_file', default=None, help='resume from file')
    parser.add_argument('--checkpoint_dir', required=True, help='Directory for storing check points')
    parser.add_argument('--aux_loss_type', default='l2', type=str, help='l2 or sgm or batchsgm')
    parser.add_argument('--aux_loss_wt', default=0.1, type=float, help='loss_wt')
    parser.add_argument('--num_classes',default=1000, type=float, help='num classes')
    parser.add_argument('--dampening', default=0, type=float, help='dampening')
    parser.add_argument('--warmup_epochs', default=0, type=int, help='iters for warmup')
    parser.add_argument('--warmup_lr', default=0.01, type=int, help='lr for warmup')

    return parser.parse_args()