aboutsummaryrefslogtreecommitdiff
path: root/examples/graph_lenet.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'examples/graph_lenet.cpp')
-rw-r--r--examples/graph_lenet.cpp3
1 files changed, 2 insertions, 1 deletions
diff --git a/examples/graph_lenet.cpp b/examples/graph_lenet.cpp
index 51b0881b1b..1427abe15f 100644
--- a/examples/graph_lenet.cpp
+++ b/examples/graph_lenet.cpp
@@ -25,6 +25,7 @@
#error "This example needs to be built with -DARM_COMPUTE_CL"
#endif /* ARM_COMPUTE_CL */
+#include "arm_compute/core/Logger.h"
#include "arm_compute/graph/Graph.h"
#include "arm_compute/graph/Nodes.h"
#include "arm_compute/runtime/CL/CLScheduler.h"
@@ -101,7 +102,7 @@ void main_graph_lenet(int argc, const char **argv)
}
Graph graph;
- graph.set_info_enablement(true);
+ arm_compute::Logger::get().set_logger(std::cout, arm_compute::LoggerVerbosity::INFO);
//conv1 << pool1 << conv2 << pool2 << fc1 << act1 << fc2 << smx
graph << hint