提交 484407e2 编写于 作者: L liuruilong

format files

上级 8159c763
...@@ -67,7 +67,7 @@ class FusionConvAddOp : public framework::OperatorWithKernel< ...@@ -67,7 +67,7 @@ class FusionConvAddOp : public framework::OperatorWithKernel<
#ifdef PADDLE_MOBILE_CPU #ifdef PADDLE_MOBILE_CPU
#ifndef CONV_ADD_REGISTER #ifndef CONV_ADD_REGISTER
static framework::FusionOpRegistrar convadd_registrar( static framework::FusionOpRegistrar convadd_registrar(
new FusionConvAddMatcher()); new FusionConvAddMatcher());
#define CONV_ADD_REGISTER #define CONV_ADD_REGISTER
#endif #endif
......
...@@ -19,8 +19,9 @@ limitations under the License. */ ...@@ -19,8 +19,9 @@ 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", // auto isok = paddle_mobile.Load(g_mobilenet_ssd_gesture + "/model",
// g_mobilenet_ssd_gesture + "/params", true); // g_mobilenet_ssd_gesture + "/params",
// true);
auto isok = paddle_mobile.Load(g_mobilenet_ssd, false); auto isok = paddle_mobile.Load(g_mobilenet_ssd, false);
if (isok) { if (isok) {
auto time2 = time(); auto time2 = time();
...@@ -36,7 +37,8 @@ int main() { ...@@ -36,7 +37,8 @@ int main() {
auto time3 = time(); auto time3 = time();
paddle_mobile.Predict(input, dims); paddle_mobile.Predict(input, dims);
auto time4 = time(); auto time4 = time();
std::cout << "predict cost :" << time_diff(time3, time4) << "ms" << std::endl; std::cout << "predict cost :" << time_diff(time3, time4) << "ms"
<< std::endl;
} }
return 0; return 0;
} }
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册