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

updating script

parent fda81a21
......@@ -29,7 +29,6 @@ python tasks/main.py \
--retriever-seq-length 256 \
--vocab-file bert-vocab.txt\
--qa-data-test ${QA_FILE} \
--num-workers 2 \
--faiss-use-gpu \
--retriever-report-topk-accuracies 1 5 20 100 \
--fp16 \
......
......@@ -36,7 +36,7 @@ python -m torch.distributed.launch $DISTRIBUTED_ARGS ./tasks/main.py \
--bert-load ${BERT_LOAD_PATH} \
--save-interval 5000 \
--log-interval 10 \
--eval-interval 25000 \
--eval-interval 250000 \
--eval-iters 100 \
--indexer-log-interval 1000 \
--faiss-use-gpu \
......
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