CUDA_VISIBLE_DEVICES=0 python infer.py --use_gpu 1 \
                                       --model_dir 'model_dir'