def parse_args()

in run_with_submitit.py [0:0]

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

def parse_args():
    parser = argparse.ArgumentParser(
        "Submitit for onestage training", add_help=False
    )
    parser.add_argument(
        "--num_gpus",
        help="Number of GPUs",
        default=8,
        type=int,
    )
    parser.add_argument(
        "--num_shards",
        help="Number of Nodes",
        default=1,
        type=int,
    )
    parser.add_argument(
        "--partition", 
        default="learnfair", 
        type=str, 
        help="Partition where to submit"
    )
    parser.add_argument(
        "--timeout", 
        default=60 * 72, 
        type=int, 
        help="Duration of the job"
    )
    parser.add_argument(
        "--cfg", 
        dest="cfg_file", 
        help="Path to the config file",
        default="configs/test_R50_8GPU.yaml", type=str
    )
    parser.add_argument(
        "--job_dir", 
        default="/checkpoint/mandelapatrick/slowfast_ssv2", 
        type=str, 
        help="Job dir. Leave empty for automatic."
    )
    parser.add_argument(
        "--name", 
        default="", 
        type=str, 
        help="Job dir. Leave empty for automatic."
    )
    parser.add_argument(
        "--resume-from",
        default="",
        type=str,
        help=(
            "Weights to resume from (.*pth file) or a file (last_checkpoint) that contains "
            + "weight file name from the same directory"
        ),
    )
    parser.add_argument(
        "--resume-job", 
        default="", 
        type=str, 
        help="resume training from the job")
    parser.add_argument(
        "--use_volta32", 
        action='store_true', 
        help="Big models? Use this")
    parser.add_argument(
        "--postfix", 
        default="experiment", 
        type=str, 
        help="Postfix of the jobs"
    )
    parser.add_argument(
        "--mail", 
        default="", 
        type=str,
        help="Email this user when the job finishes if specified"
    )
    parser.add_argument(
        '--comment', 
        default="", 
        type=str,
        help='Comment to pass to scheduler, e.g. priority message'
    )
    parser.add_argument(
        "opts",
        help="See slowfast/config/defaults.py for all options",
        default=None,
        nargs=argparse.REMAINDER,
    )
    return parser.parse_args()