diff options
Diffstat (limited to 'examples/graph_vgg16.cpp')
| -rw-r--r-- | examples/graph_vgg16.cpp | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/examples/graph_vgg16.cpp b/examples/graph_vgg16.cpp index d58bf6cbf5..2c7f614f64 100644 --- a/examples/graph_vgg16.cpp +++ b/examples/graph_vgg16.cpp @@ -65,9 +65,6 @@ public: return false; } - // Checks - ARM_COMPUTE_EXIT_ON_MSG(arm_compute::is_data_type_quantized_asymmetric(common_params.data_type), "QASYMM8 not supported for this graph"); - // Print parameter values std::cout << common_params << std::endl; @@ -224,10 +221,11 @@ public: // Finalize graph GraphConfig config; - config.num_threads = common_params.threads; - config.use_tuner = common_params.enable_tuner; - config.tuner_mode = common_params.tuner_mode; - config.tuner_file = common_params.tuner_file; + config.num_threads = common_params.threads; + config.use_tuner = common_params.enable_tuner; + config.tuner_mode = common_params.tuner_mode; + config.tuner_file = common_params.tuner_file; + config.convert_to_uint8 = (common_params.data_type == DataType::QASYMM8); graph.finalize(common_params.target, config); |