丁云鹏 5 месяцев назад
Родитель
Сommit
490cfedcb9

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

@@ -26,7 +26,9 @@ public class DNNModel implements Model {
         Predictor predictor = Predictor.clonePaddlePredictor(sourcePredictor);
         String inNames = predictor.getInputNameById(0);
         Tensor inHandle = predictor.getInputHandle(inNames);
-        log.info("predictor2 inNames={}", inNames);
+        for (int i = 0; i < predictor.getInputNum(); i++) {
+            log.info("predictor2 inNames{}={}", i, inNames);
+        }
         inHandle.reshape(2, new int[]{1, 1});
         float[] inData = new float[1 * 1];
         inHandle.copyFromCpu(inData);