فهرست منبع

model service

丁云鹏 5 ماه پیش
والد
کامیت
cd2738507c

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

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