def add_cli_args()

in benchmarks/horovod-resnet/train_imagenet_resnet_hvd.py [0:0]

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

def add_cli_args():
    cmdline = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    # Basic options
    cmdline.add_argument(
        "-m",
        "--model",
        default="resnet50",
        help="""Name of model to run: resnet[18,34,50,101,152]""",
    )
    cmdline.add_argument(
        "--data_dir",
        help="""Path to dataset in TFRecord format