Commit 4e8f8c5e authored by Mostofa Patwary's avatar Mostofa Patwary
Browse files

fixed arguments

parent 78a3dc32

megatron/arguments.py

+2 −2
Original line number Diff line number Diff line
@@ -675,8 +675,8 @@ def _add_biencoder_args(parser): 


    # training

    group.add_argument('--report-topk-accuracies', nargs='+', type=int, 

                        default=[], help='Which top-k accuracies to report '

                        '(e.g. '1 5 20')')

                        default=[], help="Which top-k accuracies to report "

                        "(e.g. '1 5 20')")

    group.add_argument('--retriever-score-scaling', action='store_true',

                       help='Whether to scale retriever scores by inverse '

                        'square root of hidden size')