diff options
Diffstat (limited to 'tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp')
-rw-r--r-- | tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp b/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp index ad79d49f0c..7396b7672c 100644 --- a/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp +++ b/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp @@ -37,6 +37,7 @@ int main(int argc, char* argv[]) modelParams.m_IsModelBinary = true; modelParams.m_ComputeDevice = modelOptions.m_ComputeDevice; modelParams.m_VisualizePostOptimizationModel = modelOptions.m_VisualizePostOptimizationModel; + modelParams.m_EnableFp16TurboMode = modelOptions.m_EnableFp16TurboMode; return std::make_unique<YoloInferenceModel>(modelParams); }); |