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

updated script as well

parent 447c1171

examples/pretrain_ict.sh

+1 −1
Original line number Diff line number Diff line
@@ -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 \