|
@@ -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();
|