|
@@ -21,8 +21,9 @@ public class DemoModel implements Model {
|
|
|
}
|
|
|
|
|
|
public String predict(String param) {
|
|
|
- Predictor predictor = Predictor.clonePaddlePredictor(sourcePredictor);
|
|
|
|
|
|
+ long time1 = System.currentTimeMillis();
|
|
|
+ Predictor predictor = Predictor.clonePaddlePredictor(sourcePredictor);
|
|
|
String inNames = predictor.getInputNameById(0);
|
|
|
Tensor inHandle = predictor.getInputHandle(inNames);
|
|
|
|
|
@@ -35,7 +36,8 @@ public class DemoModel implements Model {
|
|
|
Tensor outHandle = predictor.getOutputHandle(outNames);
|
|
|
float[] outData = new float[outHandle.getSize()];
|
|
|
outHandle.copyToCpu(outData);
|
|
|
-
|
|
|
+ long time2 = System.currentTimeMillis();
|
|
|
+ log.info("predictor2 outData[0]={},outDataLen={},cost={}", outData[0], outData.length, (time2 - time1));
|
|
|
|
|
|
outHandle.destroyNativeTensor();
|
|
|
inHandle.destroyNativeTensor();
|