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

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