From c23cb2d8d6d9f2e23db4111634b5583e9c819d31 Mon Sep 17 00:00:00 2001
From: zhangyang <zhangyang49@baidu.com>
Date: Wed, 15 Aug 2018 14:10:44 +0800
Subject: [PATCH] recover test-googlenet

---
 test/net/test_googlenet.cpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/test/net/test_googlenet.cpp b/test/net/test_googlenet.cpp
index 202f6d2898..b98d07ad16 100644
--- a/test/net/test_googlenet.cpp
+++ b/test/net/test_googlenet.cpp
@@ -28,16 +28,16 @@ int main() {
   paddle_mobile.SetThreadNum(4);
   bool optimize = true;
   auto time1 = time();
-  if (paddle_mobile.Load(g_resnet, optimize)) {
+  if (paddle_mobile.Load(g_googlenet, optimize)) {
     auto time2 = time();
     std::cout << "load cost :" << time_diff(time1, time2) << "ms" << std::endl;
     std::vector<float> input;
     std::vector<int64_t> dims{1, 3, 224, 224};
-    // GetInput<float>(g_test_image_1x3x224x224, &input, dims);
+    GetInput<float>(g_test_image_1x3x224x224, &input, dims);
     // 预热一次
     auto vec_result = paddle_mobile.Predict(input, dims);
     auto time3 = time();
-    for (int i = 0; i < 1; ++i) {
+    for (int i = 0; i < 10; ++i) {
       auto vec_result = paddle_mobile.Predict(input, dims);
     }
     auto time4 = time();
-- 
GitLab