From 04c3a3e67c5d51156a45ed370d9f21cc8d764631 Mon Sep 17 00:00:00 2001 From: liuruilong Date: Wed, 18 Jul 2018 19:35:47 +0800 Subject: [PATCH] add loss .h --- src/operators/math/im2col.cpp | 2 +- test/net/test_mobilenet+ssd.cpp | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/operators/math/im2col.cpp b/src/operators/math/im2col.cpp index e4d0c816d7..7b0b974b54 100644 --- a/src/operators/math/im2col.cpp +++ b/src/operators/math/im2col.cpp @@ -15,7 +15,7 @@ limitations under the License. */ #include "operators/math/im2col.h" #include #ifdef __ARM_NEON -#include +#include #endif #include "common/types.h" namespace paddle_mobile { diff --git a/test/net/test_mobilenet+ssd.cpp b/test/net/test_mobilenet+ssd.cpp index f16f91ebbe..051d5f343b 100644 --- a/test/net/test_mobilenet+ssd.cpp +++ b/test/net/test_mobilenet+ssd.cpp @@ -19,8 +19,10 @@ limitations under the License. */ int main() { paddle_mobile::PaddleMobile paddle_mobile; auto time1 = time(); - - if (paddle_mobile.Load(g_mobilenet_ssd, false)) { +// auto isok = paddle_mobile.Load(g_mobilenet_ssd_gesture + "/model", +// g_mobilenet_ssd_gesture + "/params", true); + auto isok = paddle_mobile.Load(g_mobilenet_ssd, false); + if (isok) { auto time2 = time(); DLOG << "load cost :" << time_diff(time1, time1) << "ms"; -- GitLab