Skip to content

Deploy alt_gvt_base model will raise matmul_v2 error by using mkldnn. #36217

Closed
@baoachun

Description

@baoachun

图片

git clone https://github.com/PaddlePaddle/PaddleClas.git
cd PaddleClas;mkdir model
cd model;wget https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/alt_gvt_base_pretrained.pdparams
cd ..
python3 tools/export_model.py -c ppcls/configs/ImageNet/Twins/alt_gvt_base.yaml -o Global.pretrained_model=./model/alt_gvt_base_pretrained
cd deploy; python3 python/predict_cls.py -c configs/inference_cls.yaml -o Global.inference_model_dir=../inference -o Global.batch_size=1 -o Global.use_gpu=False -o Global.use_tensorrt=False -o Global.enable_mkldnn=True

Metadata

Metadata

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions