Browse Source

model service

丁云鹏 5 months ago
parent
commit
8797a095d7

+ 4 - 10
recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/service/model/DemoModel.java

@@ -53,18 +53,12 @@ public class DemoModel implements Model {
             String modelDir = PropertiesUtil.getString("model.dir") + "/demo";
             CompressUtil.decompressGzFile(in, modelDir);
 
-            String modelFile = modelDir + "/inference.pdmodel";
-            String paramFile = modelDir + "/inference.pdiparams";
-            File file = new File(modelFile);
-            if (file.exists()) {
-                log.info("modelFile exists {}", file.getAbsolutePath());
-            } else {
-                log.info("modelFile not exists {}", file.getAbsolutePath());
-            }
+            File modelFile = new File(modelDir + "/inference.pdmodel");
+            File paramFile = new File(modelDir + "/inference.pdiparams");
 
             log.info("start init Config");
             Config config = new Config();
-            config.setCppModel(modelFile, paramFile);
+            config.setCppModel(modelFile.getAbsolutePath(), paramFile.getAbsolutePath());
             config.enableMemoryOptim(true);
             config.enableProfile();
             config.enableMKLDNN();
@@ -74,7 +68,7 @@ public class DemoModel implements Model {
 
             log.info("start init Predictor");
             Predictor predictor = Predictor.createPaddlePredictor(config);
-            Predictor temp = predictor;
+            Predictor temp = sourcePredictor;
             this.sourcePredictor = predictor;
             temp.destroyNativePredictor();
             return true;