|
@@ -26,7 +26,9 @@ public class DNNModel implements Model {
|
|
Predictor predictor = Predictor.clonePaddlePredictor(sourcePredictor);
|
|
Predictor predictor = Predictor.clonePaddlePredictor(sourcePredictor);
|
|
String inNames = predictor.getInputNameById(0);
|
|
String inNames = predictor.getInputNameById(0);
|
|
Tensor inHandle = predictor.getInputHandle(inNames);
|
|
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});
|
|
inHandle.reshape(2, new int[]{1, 1});
|
|
float[] inData = new float[1 * 1];
|
|
float[] inData = new float[1 * 1];
|
|
inHandle.copyFromCpu(inData);
|
|
inHandle.copyFromCpu(inData);
|