From 344081db43bd9b6335c103c23a419de351aee88c Mon Sep 17 00:00:00 2001
From: ShiningZhang <zhang_liang1991@126.com>
Date: Mon, 20 Dec 2021 17:44:59 +0800
Subject: [PATCH] add suffix of model&params in paddle_engine

---
 paddle_inference/paddle/include/paddle_engine.h | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle_inference/paddle/include/paddle_engine.h b/paddle_inference/paddle/include/paddle_engine.h
index 95205610..6a673be0 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<std::string> suffixParaVector = {".pdiparams", "__params__"};
-    std::vector<std::string> suffixModelVector = {".pdmodel", "__model__"};
+    std::vector<std::string> suffixParaVector = {".pdiparams", "__params__", "params"};
+    std::vector<std::string> suffixModelVector = {".pdmodel", "__model__", "model"};
     std::string paraFileName = getFileBySuffix(model_path, suffixParaVector);
     std::string modelFileName = getFileBySuffix(model_path, suffixModelVector);
 
-- 
GitLab