diff --git a/test/net/test_googlenet.cpp b/test/net/test_googlenet.cpp index 202f6d28983c36579bee29d3190a9ec36e83251c..b98d07ad16dcb15268e4638f9144bde36a1005a8 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 input; std::vector dims{1, 3, 224, 224}; - // GetInput(g_test_image_1x3x224x224, &input, dims); + GetInput(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();