丁云鹏 5 달 전
부모
커밋
7581a2fd11
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/service/model/DNNModel.java

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

@@ -53,7 +53,7 @@ public class DNNModel implements Model {
         CompressUtil.decompressGzFile(in, modelDir);
 
         String modelFile = modelDir + "/0/rec_static.pdmodel";
-        String paramFile = modelDir + "/0/rec_static.pdiparams";
+        String paramFile = modelDir + "/0/rec_static.pdparams";
 
         Config config = new Config();
         config.setCppModel(modelFile, paramFile);