dnn test

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

@@ -31,7 +31,7 @@ public class DNNModel implements Model {
         }
 
         for (int i = 0; i < predictor.getOutputNum(); i++) {
-            log.info("predictor2 outName{}={}", i, predictor.getInputNameById(i));
+            log.info("predictor2 outName{}={}", i, predictor.getOutputNameById(i));
         }
 
         inHandle.reshape(2, new int[]{1, 1});

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

@@ -32,7 +32,7 @@ public class DemoModel implements Model {
         }
 
         for (int i = 0; i < predictor.getOutputNum(); i++) {
-            log.info("predictor2 outName{}={}", i, predictor.getInputNameById(i));
+            log.info("predictor2 outName{}={}", i, predictor.getOutputNameById(i));
         }
         inHandle.reshape(4, new int[]{1, 3, 224, 224});