fixed arguments

4e8f8c5e · Mostofa Patwary · 78a3dc32 · 4e8f8c5e
Commit 4e8f8c5e authored Feb 03, 2021 by Mostofa Patwary
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

megatron/arguments.py megatron/arguments.py +2 -2

No files found.
--- a/megatron/arguments.py
+++ b/megatron/arguments.py
@@ -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 '
+                        default=[], help="Which top-k accuracies to report "
-                        '(e.g. '1 5 20')')
+                        "(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')