提交 d58480ea 编写于 作者: L liuruilong

add loss .h

上级 f1bf0b94
......@@ -15,7 +15,7 @@ limitations under the License. */
#include "operators/math/im2col.h"
#include <vector>
#ifdef __ARM_NEON
#include <arm_neon>
#include <arm_neon.h>
#endif
#include "common/types.h"
namespace paddle_mobile {
......
......@@ -19,8 +19,10 @@ limitations under the License. */
int main() {
paddle_mobile::PaddleMobile<paddle_mobile::CPU> 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";
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册