inference.sh

CUDA_VISIBLE_DEVICES=1 torchrun \
     --nnodes=1 \
     --nproc_per_node=1 \
     --master_port=21547 \
     inference.py \
     -conf ./config/test.yml