|
@@ -9,8 +9,9 @@ public class test {
|
|
|
}
|
|
|
|
|
|
public static void main(String[] args) {
|
|
|
- Config config = new Config();
|
|
|
|
|
|
+ long time1 = System.currentTimeMillis();
|
|
|
+ Config config = new Config();
|
|
|
config.setCppModel(args[0], args[1]);
|
|
|
config.enableMemoryOptim(true);
|
|
|
config.enableProfile();
|
|
@@ -40,15 +41,12 @@ public class test {
|
|
|
|
|
|
predictor.tryShrinkMemory();
|
|
|
predictor.clearIntermediateTensor();
|
|
|
-
|
|
|
- System.out.println("predictor1: " + outData[0]);
|
|
|
+ long time2 = System.currentTimeMillis();
|
|
|
+ System.out.println("predictor1: " + outData[0] + ",time:" + (time2 - time1));
|
|
|
System.out.println("predictor1: " + outData.length);
|
|
|
|
|
|
test(predictor);
|
|
|
-
|
|
|
- outHandle.destroyNativeTensor();
|
|
|
- inHandle.destroyNativeTensor();
|
|
|
- predictor.destroyNativePredictor();
|
|
|
+ test(predictor);
|
|
|
|
|
|
Config newConfig = new Config();
|
|
|
newConfig.setCppModelDir("/model_dir");
|
|
@@ -62,8 +60,10 @@ public class test {
|
|
|
|
|
|
}
|
|
|
|
|
|
- private static void test(Predictor predictor) {
|
|
|
- Predictor predictor2 = Predictor.clonePaddlePredictor(predictor);
|
|
|
+ private static void test(Predictor predictor2) {
|
|
|
+
|
|
|
+ long time1 = System.currentTimeMillis();
|
|
|
+ Predictor predictor = Predictor.clonePaddlePredictor(predictor2);
|
|
|
String inNames = predictor.getInputNameById(0);
|
|
|
Tensor inHandle = predictor.getInputHandle(inNames);
|
|
|
inHandle.reshape(4, new int[]{1, 3, 224, 224});
|
|
@@ -78,7 +78,8 @@ public class test {
|
|
|
predictor.tryShrinkMemory();
|
|
|
predictor.clearIntermediateTensor();
|
|
|
|
|
|
- System.out.println("predictor2: " + outData[0]);
|
|
|
+ long time2 = System.currentTimeMillis();
|
|
|
+ System.out.println("predictor2: " + outData[0] + ",time:" + (time2 - time1));
|
|
|
System.out.println("predictor2: " + outData.length);
|
|
|
outHandle.destroyNativeTensor();
|
|
|
inHandle.destroyNativeTensor();
|