丁云鹏 5 meses atrás
pai
commit
0b10f49228

+ 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();