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