Skip to content

GitLab

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

updating script

parent fda81a21
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
examples/evaluate_retriever_nq.sh
View file @ c7c65bbb
...@@ -29,7 +29,6 @@ python tasks/main.py \ ...@@ -29,7 +29,6 @@ python tasks/main.py \
--retriever-seq-length 256 \ --retriever-seq-length 256 \
--vocab-file bert-vocab.txt\ --vocab-file bert-vocab.txt\
--qa-data-test ${QA_FILE} \ --qa-data-test ${QA_FILE} \
--num-workers 2 \
--faiss-use-gpu \ --faiss-use-gpu \
--retriever-report-topk-accuracies 1 5 20 100 \ --retriever-report-topk-accuracies 1 5 20 100 \
--fp16 \ --fp16 \
......
examples/finetune_retriever_distributed.sh
View file @ c7c65bbb
...@@ -36,7 +36,7 @@ python -m torch.distributed.launch $DISTRIBUTED_ARGS ./tasks/main.py \ ...@@ -36,7 +36,7 @@ python -m torch.distributed.launch $DISTRIBUTED_ARGS ./tasks/main.py \
--bert-load ${BERT_LOAD_PATH} \ --bert-load ${BERT_LOAD_PATH} \
--save-interval 5000 \ --save-interval 5000 \
--log-interval 10 \ --log-interval 10 \
--eval-interval 25000 \ --eval-interval 250000 \
--eval-iters 100 \ --eval-iters 100 \
--indexer-log-interval 1000 \ --indexer-log-interval 1000 \
--faiss-use-gpu \ --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