提交 04c3a3e6 编写于 作者: L liuruilong

add loss .h

上级 486460b1
...@@ -15,7 +15,7 @@ limitations under the License. */ ...@@ -15,7 +15,7 @@ limitations under the License. */
#include "operators/math/im2col.h" #include "operators/math/im2col.h"
#include <vector> #include <vector>
#ifdef __ARM_NEON #ifdef __ARM_NEON
#include <arm_neon> #include <arm_neon.h>
#endif #endif
#include "common/types.h" #include "common/types.h"
namespace paddle_mobile { namespace paddle_mobile {
......
...@@ -19,8 +19,10 @@ limitations under the License. */ ...@@ -19,8 +19,10 @@ limitations under the License. */
int main() { int main() {
paddle_mobile::PaddleMobile<paddle_mobile::CPU> paddle_mobile; paddle_mobile::PaddleMobile<paddle_mobile::CPU> paddle_mobile;
auto time1 = time(); auto time1 = time();
// auto isok = paddle_mobile.Load(g_mobilenet_ssd_gesture + "/model",
if (paddle_mobile.Load(g_mobilenet_ssd, false)) { // g_mobilenet_ssd_gesture + "/params", true);
auto isok = paddle_mobile.Load(g_mobilenet_ssd, false);
if (isok) {
auto time2 = time(); auto time2 = time();
DLOG << "load cost :" << time_diff(time1, time1) << "ms"; DLOG << "load cost :" << time_diff(time1, time1) << "ms";
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册