def METHOD_NAME(parser):
    """Args for MaskedLM Loss"""
    # Default for masked_lm_only is False so as to not break BERT training
    parser.add_argument(
        "--masked-lm-only",
        default=False,
        action="store_true",
        help="compute MLM loss only",
    )
    parser.add_argument(
        "--nsp-loss-weight",
        default=1.0,
        type=float,
        help="weight for next sentence prediction" " loss (default 1)",
    )