eval_doc.sh 379 Bytes
Newer Older
Melos's avatar
Melos committed
1
2
3
4
5
6
7
8
9
10

EVAL_PTH=$1
SAVE_NAME=$2




python -m torch.distributed.launch --use-env     --nproc_per_node ${NPROC_PER_NODE:-8}     --nnodes ${WORLD_SIZE:-1}     --node_rank ${RANK:-0} --master_addr ${MASTER_ADDR:-127.0.0.1}     --master_port ${MASTER_PORT:-12345}     eval/evaluate_vqa_doc.py     --checkpoint $EVAL_PTH         --batch-size 8      --num-workers 4 --save_name $SAVE_NAME