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