From 800925760890d896b73ea3b2e2e5fa18c02fd5b0 Mon Sep 17 00:00:00 2001 From: Wangzheee <634486483@qq.com> Date: Mon, 9 Aug 2021 15:52:27 +0800 Subject: [PATCH] fix_trt_int8 (#34704) --- paddle/fluid/inference/tensorrt/engine.cc | 1 - 1 file changed, 1 deletion(-) diff --git a/paddle/fluid/inference/tensorrt/engine.cc b/paddle/fluid/inference/tensorrt/engine.cc index d9e87122ac2..dbaaf2bdc7c 100644 --- a/paddle/fluid/inference/tensorrt/engine.cc +++ b/paddle/fluid/inference/tensorrt/engine.cc @@ -89,7 +89,6 @@ void TensorRTEngine::FreezeNetwork() { if (enable_int8) { infer_builder_config_->setFlag(nvinfer1::BuilderFlag::kFP16); infer_builder_config_->setFlag(nvinfer1::BuilderFlag::kINT8); - infer_builder_config_->setFlag(nvinfer1::BuilderFlag::kSTRICT_TYPES); if (calibrator_) { infer_builder_config_->setInt8Calibrator(calibrator_); -- GitLab