def add_model_specific_args()

in cp_examples/mip_finetune/mip_model.py [0:0]

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

    def add_model_specific_args(parent_parser):
        parser = ArgumentParser(parents=[parent_parser], add_help=False)
        parser.add_argument("--pretrained_file", type=Path, required=True)
        parser.add_argument("--arch", default="densenet121", type=str)
        parser.add_argument("--num_classes", default=14, type=int)
        parser.add_argument("--val_pathology_list", nargs="+")
        parser.add_argument("--pos_weights", default=None, type=float)
        # Training params
        parser.add_argument("--learning_rate", default=1e-3, type=float)
        parser.add_argument("--epochs", default=50, type=int)
        # Model params
        parser.add_argument("--projection_dim", type=int, default=64)
        parser.add_argument("--num_heads", type=int, default=2)
        parser.add_argument("--feedforward_dim", type=int, default=128)
        parser.add_argument("--drop_transformer", type=float, default=0.5)
        parser.add_argument("--drop_cpe", type=float, default=0.5)
        parser.add_argument(
            "--pooling", choices=["last_timestep", "sum"], default="last_timestep"
        )
        parser.add_argument("--image_shape", default=(7, 7))
        return parser