丁云鹏 пре 6 месеци
родитељ
комит
e2841bfd11

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

@@ -27,13 +27,6 @@ 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.getOutputNameById(i));
-        }
         inHandle.reshape(4, new int[]{1, 3, 224, 224});
 
         float[] inData = new float[1 * 3 * 224 * 224];
@@ -64,7 +57,6 @@ public class DemoModel implements Model {
         Config config = new Config();
         config.setCppModel(modelFile, paramFile);
         config.enableMemoryOptim(true);
-        config.enableProfile();
         config.enableMKLDNN();
         config.getCpuMathLibraryNumThreads();
         config.getFractionOfGpuMemoryForPool();