Tingquan Gao 11357c0fd1
Modify the "cls_model_dir" to "cls_model_path" and "cls_params_path" (#411)
* Modify the "cls_model_dir" to "cls_model_path" and "cls_params_path"
2020-11-19 15:00:17 +08:00

13 lines
265 B
Plaintext
Executable File

# model load config
use_gpu 0
gpu_id 0
gpu_mem 4000
cpu_math_library_num_threads 10
use_mkldnn 1
# cls config
cls_model_path /PaddleClas/inference/cls_infer.pdmodel
cls_params_path /PaddleClas/inference/cls_infer.pdiparams
resize_short_size 256
crop_size 224