From 347655b9cc676eab90182f54ee4a04460b9162ce Mon Sep 17 00:00:00 2001 From: zhaojiaying01 Date: Thu, 23 Aug 2018 14:06:51 +0800 Subject: [PATCH] update unit test and jni --- test/net/test_mobilenet+ssd.cpp | 6 +++--- test/net/test_mobilenet.cpp | 6 +++--- test/net/test_squeezenet.cpp | 6 +++--- test/net/test_yolo.cpp | 6 +++--- 4 files changed, 12 insertions(+), 12 deletions(-) diff --git a/test/net/test_mobilenet+ssd.cpp b/test/net/test_mobilenet+ssd.cpp index c7ba847f7e..0bb6fea0e7 100644 --- a/test/net/test_mobilenet+ssd.cpp +++ b/test/net/test_mobilenet+ssd.cpp @@ -33,9 +33,9 @@ int main() { GetInput(g_hand, &input, dims); // 预热一次 - for (int i = 0; i < 10; ++i) { - auto output = paddle_mobile.Predict(input, dims); - } + for (int i = 0; i < 10; ++i) { + auto output = paddle_mobile.Predict(input, dims); + } auto time3 = time(); for (int i = 0; i < 10; ++i) { auto output = paddle_mobile.Predict(input, dims); diff --git a/test/net/test_mobilenet.cpp b/test/net/test_mobilenet.cpp index 3fd9ee320b..fe7b431caf 100644 --- a/test/net/test_mobilenet.cpp +++ b/test/net/test_mobilenet.cpp @@ -39,9 +39,9 @@ int main() { std::cout << " Max element is " << *biggest << " at position " << std::distance(std::begin(vec_result), biggest) << std::endl; - for (int i = 0; i < 10; ++i) { - auto vec_result = paddle_mobile.Predict(input, dims); - } + for (int i = 0; i < 10; ++i) { + auto vec_result = paddle_mobile.Predict(input, dims); + } auto time3 = time(); for (int i = 0; i < 10; ++i) { auto vec_result = paddle_mobile.Predict(input, dims); diff --git a/test/net/test_squeezenet.cpp b/test/net/test_squeezenet.cpp index 7a5022b356..3ed0299d9d 100644 --- a/test/net/test_squeezenet.cpp +++ b/test/net/test_squeezenet.cpp @@ -33,9 +33,9 @@ int main() { std::vector input(input_tensor.data(), input_tensor.data() + input_tensor.numel()); // 预热一次 - for (int i = 0; i < 10; ++i) { - paddle_mobile.Predict(input, dims); - } + for (int i = 0; i < 10; ++i) { + paddle_mobile.Predict(input, dims); + } auto time3 = time(); for (int i = 0; i < 10; ++i) { paddle_mobile.Predict(input, dims); diff --git a/test/net/test_yolo.cpp b/test/net/test_yolo.cpp index 15bb5ea321..bb566d8b8e 100644 --- a/test/net/test_yolo.cpp +++ b/test/net/test_yolo.cpp @@ -34,9 +34,9 @@ int main() { std::vector input(input_tensor.data(), input_tensor.data() + input_tensor.numel()); // 预热一次 - for (int i = 0; i < 10; ++i) { - paddle_mobile.Predict(input, dims); - } + for (int i = 0; i < 10; ++i) { + paddle_mobile.Predict(input, dims); + } auto time3 = time(); for (int i = 0; i < 10; ++i) { paddle_mobile.Predict(input, dims); -- GitLab