丁云鹏 1 سال پیش
والد
کامیت
d54c55b11d
1فایلهای تغییر یافته به همراه2 افزوده شده و 2 حذف شده
  1. 2 2
      recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/service/model/DNNModel.java

+ 2 - 2
recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/service/model/DNNModel.java

@@ -52,8 +52,8 @@ public class DNNModel implements Model {
         String modelDir = PropertiesUtil.getString("model.dir") + "/dnn";
         String modelDir = PropertiesUtil.getString("model.dir") + "/dnn";
         CompressUtil.decompressGzFile(in, modelDir);
         CompressUtil.decompressGzFile(in, modelDir);
 
 
-        String modelFile = modelDir + "/inference.pdmodel";
-        String paramFile = modelDir + "/inference.pdiparams";
+        String modelFile = modelDir + "/rec_static.pdmodel";
+        String paramFile = modelDir + "/rec_static.pdiparams";
 
 
         Config config = new Config();
         Config config = new Config();
         config.setCppModel(modelFile, paramFile);
         config.setCppModel(modelFile, paramFile);