diff options
Diffstat (limited to 'examples/graph_lenet.cpp')
-rw-r--r-- | examples/graph_lenet.cpp | 3 |
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 |