7d40903e021cbf8893feeef0a7654aab53ef06c2,catalyst/contrib/scripts/check_index_model.py,,parse_args,#,8

Before Change

        default="l2",
        choices=["l2", "angulardist", "cosinesimil"]
    )
    parser.add_argument(
        "-b",
        "--batch-size",
        default=128,
        type=int,
        metavar="N",
        help="mini-batch size "
    )
    parser.add_argument("-k", "--recall-at", default="1,3,5,10", type=str)

    return parser.parse_args()

After Change

    parser = argparse.ArgumentParser()
    build_args(parser)
    args = parser.parse_args()
    return args


def main(args, _=None):
    print("[==       Loading features       ==]")

Instances