dnn test

recommend-model-produce/src/main/python/models/dnn/criteo_reader.py

@@ -75,6 +75,9 @@ class RecDataset(IterableDataset):
                     for key, value in output[:-1]:
                         output_list.append(np.array(value).astype('int64'))
                     # dense
+                    
+                    print("output_list", output_list)
+
                     output_list.append(
                         np.array(output[-1][1]).astype("float32"))
                     # list

recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/service/model/DemoModel.java

@@ -27,7 +27,13 @@ public class DemoModel implements Model {
         Predictor predictor = Predictor.clonePaddlePredictor(sourcePredictor);
         String inNames = predictor.getInputNameById(0);
         Tensor inHandle = predictor.getInputHandle(inNames);
+        for (int i = 0; i < predictor.getInputNum(); i++) {
+            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(4, new int[]{1, 3, 224, 224});
 
         float[] inData = new float[1 * 3 * 224 * 224];