diff --git a/test/net/test_yolo.cpp b/test/net/test_yolo.cpp index 700eb10cac6f0b80595d8c53866c7f675d2b56fb..40aabe92f1bff84388ddc411b8517536d33ddb01 100644 --- a/test/net/test_yolo.cpp +++ b/test/net/test_yolo.cpp @@ -22,9 +22,10 @@ int main() { // ../../../test/models/googlenet // ../../../test/models/mobilenet auto time1 = time(); - if (paddle_mobile.Load(g_yolo, true)) { + if (paddle_mobile.Load(std::string(g_yolo) + "/model", + std::string(g_yolo) + "/params", true)) { auto time2 = time(); - std::cout << "load cost :" << time_diff(time1, time1) << "ms" << std::endl; + std::cout << "load cost :" << time_diff(time1, time2) << "ms" << std::endl; std::vector dims{1, 3, 227, 227}; Tensor input_tensor;