File size: 271 Bytes
7134ce7
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
bash scripts/env.sh
CKPT=$1
OUTPUT=$2
torchrun --nnodes=1 \
         --node_rank=0 \
         --master_addr=127.0.0.1 \
         --nproc_per_node=1 \
         --master_port=63668 \
         evaluation/inference_withtool.py --checkpoint $CKPT --output_name $OUTPUT