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

updating script

parent fda81a21

examples/evaluate_retriever_nq.sh

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

examples/finetune_retriever_distributed.sh

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