def get_args()

in src/sagemaker_defect_detection/classifier.py [0:0]

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

def get_args() -> Namespace:
    parent_parser = ArgumentParser(add_help=False)
    aa = parent_parser.add_argument
    aa("--epochs", type=int, default=100, help="number of training epochs")
    aa("--save-path", metavar="DIR", default=os.getenv("SM_MODEL_DIR", ""), type=str, help="path to save output")
    aa("--gpus", type=int, default=os.getenv("SM_NUM_GPUS", 1), help="how many gpus")
    aa(
        "--distributed-backend",
        type=str,
        default="",
        choices=("dp", "ddp", "ddp2"),
        help="supports three options dp, ddp, ddp2",
    )
    aa("--use-16bit", dest="use_16bit", action="store_true", help="if true uses 16 bit precision")

    parser = DDNClassification.add_model_specific_args(parent_parser)
    return parser.parse_args()