丁云鹏 5 月之前
父节点
当前提交
1b5a21c2ac

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

@@ -27,8 +27,13 @@ public class DNNModel implements Model {
         String inNames = predictor.getInputNameById(0);
         Tensor inHandle = predictor.getInputHandle(inNames);
         for (int i = 0; i < predictor.getInputNum(); i++) {
-            log.info("predictor2 inNames{}={}", i, inNames);
+            log.info("predictor2 inName{}={}", i, predictor.getInputNameById(i));
         }
+
+        for (int i = 0; i < predictor.getOutputNum(); i++) {
+            log.info("predictor2 outName{}={}", i, predictor.getInputNameById(i));
+        }
+
         inHandle.reshape(2, new int[]{1, 1});
         float[] inData = new float[1 * 1];
         inHandle.copyFromCpu(inData);