diff --git a/paddle_inference/paddle/include/paddle_engine.h b/paddle_inference/paddle/include/paddle_engine.h index 6e79f771c25651ae374b1845c6307bd8654d13c7..14fa447c6fd18b1e6a36b5106ed2851125a00823 100644 --- a/paddle_inference/paddle/include/paddle_engine.h +++ b/paddle_inference/paddle/include/paddle_engine.h @@ -166,8 +166,8 @@ class PaddleInferenceEngine : public EngineCore { } Config config; - std::vector suffixParaVector = {".pdiparams", "__params__"}; - std::vector suffixModelVector = {".pdmodel", "__model__"}; + std::vector suffixParaVector = {".pdiparams", "__params__", "params"}; + std::vector suffixModelVector = {".pdmodel", "__model__", "model"}; std::string paraFileName = getFileBySuffix(model_path, suffixParaVector); std::string modelFileName = getFileBySuffix(model_path, suffixModelVector);