소스 검색

save model

丁云鹏 5 달 전
부모
커밋
0b10f49228
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

@@ -34,8 +34,8 @@ public class DNNModel implements Model {
             log.info("predictor2 outName{}={}", i, predictor.getOutputNameById(i));
         }
 
-        inHandle.reshape(2, new int[]{1, 1});
-        float[] inData = new float[1 * 1];
+        inHandle.reshape(2, new int[]{1, 158});
+        float[] inData = new float[1 * 158];
         inHandle.copyFromCpu(inData);
         predictor.run();