def METHOD_NAME(parser):
    TransformerModel.METHOD_NAME(parser)
    # encoder head selection
    parser.add_argument(
        "--encoder-attn-head-select",
        action="store_true",
        default=False,
        help="encoder head selection"
    )
    parser.add_argument(
        "--total-encoder-attention-heads",
        type=int,
        help="total number of encoder attention heads"
    )
    # decoder self attention
    parser.add_argument(
        "--decoder-self-attn-head-select",
        action="store_true",
        default=False,
        help="decoder self-attention head selection"
    )
    # decoder-encoder attention
    parser.add_argument(
        "--dec-enc-attn-head-select",
        action="store_true",
        default=False,
        help="decoder-encoder attention head selection"
    )
    parser.add_argument(
        "--total-decoder-attention-heads",
        type=int,
        help="total number of decoder attention heads"
    )
    # selection strategy
    parser.add_argument(
        "--attn-head-select-strategy",
        type=str,
        help="attention head selection strategy, subset or group"
    )