def train_network_parse_args()

in contactopt/arguments.py [0:0]

• 22 lines of code
• 4 McCabe index (conditional complexity)

def train_network_parse_args():
    parser = argparse.ArgumentParser(description='Alignment networks training')
    parser.add_argument('--lr', default=0.01, type=float)
    parser.add_argument('--batch_size', default=32, type=int)
    parser.add_argument('--optimizer', default='adam', type=str)
    parser.add_argument('--split', default='aug', type=str)
    # parser.add_argument('--loss_pose', default=0, type=float)
    parser.add_argument('--loss_c_obj', default=1, type=float)
    parser.add_argument('--loss_c_hand', default=1, type=float)
    # parser.add_argument('--loss_3d', default=0, type=float)
    parser.add_argument('--epochs', default=101, type=int)
    parser.add_argument('--checkpoint', default='', type=str)
    parser.add_argument('--desc', default='', type=str)
    parser.add_argument('--vis', action='store_true')
    args = parser.parse_args()

    if args.desc == '':
        args.desc = str(datetime.datetime.now().strftime("%Y%m%d-%H%M%S"))

    all_str = ''
    for key, val in vars(args).items():
        all_str += '--{}={} '.format(key, val)

    print(all_str)   # Convert to dict and print
    args.all_str = all_str

    parse_dataset(args)

    return args