INNER CODE UNIT · Python

parse_args

digantamisra98/Mish · PyTorch Benchmarks/train_imagenet.py:30

def parse_args():
    parser = argparse.ArgumentParser(description="PyTorch ImageNet Training")
    parser.add_argument("data", metavar="DIR", help="path to dataset")
    parser.add_argument(
        "--arch",
        "-a",
        metavar="ARCH",
        default="resnet",
        help="model architecture: " + " | ".join(model_names) + " (default: resnet18)",
    )
    parser.add_argument(
        "--depth", default=50, type=int, metavar="D", help="model depth"
    )
    parser.add_argument(
        "--ngpu", default=4, type=int, metavar="G", help="number of gpus to use"
    )
    parser.add_argument(
        "-j",

View source record →