def METHOD_NAME(parser):
    TransformerModel.METHOD_NAME(parser)
    parser.add_argument(
        "--apply-bert-init",
        action="store_true",
        help="use custom param initialization for BERT",
    )