infer_gpu.sh 115 字节
Newer Older
O
overlordmax 已提交
1
CUDA_VISIBLE_DEVICES=0 python infer.py --use_gpu 1 \
O
overlordmax 已提交
2
                                       --model_dir ./model_dir