丁云鹏 il y a 5 mois
Parent
commit
ea7b019873

+ 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";
         CompressUtil.decompressGzFile(in, modelDir);
 
-        String modelFile = modelDir + "/rec_static.pdmodel";
-        String paramFile = modelDir + "/rec_static.pdiparams";
+        String modelFile = modelDir + "/0/rec_static.pdmodel";
+        String paramFile = modelDir + "/0/rec_static.pdiparams";
 
         Config config = new Config();
         config.setCppModel(modelFile, paramFile);

+ 1 - 1
recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/service/model/ModelEnum.java

@@ -5,7 +5,7 @@ import org.apache.commons.lang3.StringUtils;
 public enum ModelEnum {
     VIDEO_DSSM("videoDssm", "", DSSMModel.class),
     DEMO("demo", "zhangbo/model_paddle_demo.tar.gz", DemoModel.class),
-    DNN("dnn", "dyp/0/dnn.tar.gz", DNNModel.class),
+    DNN("dnn", "dyp/dnn.tar.gz", DNNModel.class),
     NULL("null", "null", null);
 
     private String modelName;