Commit a4b628ab authored by Mostofa Patwary's avatar Mostofa Patwary
Browse files

updated script as well

parent 447c1171
...@@ -27,7 +27,7 @@ python pretrain_ict.py \ ...@@ -27,7 +27,7 @@ python pretrain_ict.py \
--log-interval 100 \ --log-interval 100 \
--eval-interval 1000 \ --eval-interval 1000 \
--eval-iters 10 \ --eval-iters 10 \
--report-topk-accuracies 1 5 10 20 100 \ --retriever-report-topk-accuracies 1 5 10 20 100 \
--retriever-score-scaling \ --retriever-score-scaling \
--load $CHECKPOINT_PATH \ --load $CHECKPOINT_PATH \
--save $CHECKPOINT_PATH \ --save $CHECKPOINT_PATH \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment