def add_model_specific_args()

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

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

    def add_model_specific_args(parent_parser):  # pragma: no-cover
        parser = ArgumentParser(parents=[parent_parser], add_help=False)
        aa = parser.add_argument
        aa(
            "--data-path",
            metavar="DIR",
            type=str,
            default=os.getenv("SM_CHANNEL_TRAINING", ""),
        )
        aa(
            "--backbone",
            default="resnet34",
        )
        aa(
            "--freeze-backbone",
            action="store_true",
        )
        aa(
            "--num-classes",
            default=6,
            type=int,
            metavar="N",
        )
        aa(
            "-b",
            "--batch-size",
            default=64,
            type=int,
            metavar="N",
        )
        aa(
            "--lr",
            "--learning-rate",
            default=1e-3,
            type=float,
            metavar="LR",
            dest="learning_rate",
        )
        aa("--momentum", default=0.9, type=float, metavar="M", help="momentum")
        aa(
            "--wd",
            "--weight-decay",
            default=1e-4,
            type=float,
            metavar="W",
            dest="weight_decay",
        )
        aa(
            "--seed",
            type=int,
            default=42,
        )
        return parser