updated script as well

a4b628ab · Mostofa Patwary · 447c1171 · a4b628ab
Commit a4b628ab authored Feb 15, 2021 by Mostofa Patwary
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/pretrain_ict.sh examples/pretrain_ict.sh +1 -1

No files found.
--- a/examples/pretrain_ict.sh
+++ b/examples/pretrain_ict.sh
@@ -27,7 +27,7 @@ python pretrain_ict.py \
        --log-interval 100 \
        --eval-interval 1000 \
        --eval-iters 10 \
-        --report-topk-accuracies 1 5 10 20 100 \
+        --retriever-report-topk-accuracies 1 5 10 20 100 \
        --retriever-score-scaling \
        --load $CHECKPOINT_PATH \
        --save $CHECKPOINT_PATH \