fix_trt_int8 (#34704)

80092576 · Wangzheee · GitHub · 7afd31bb · 80092576
显示空白变更内容
内联并排

Showing with 0 addition and 1 deletion

paddle/fluid/inference/tensorrt/engine.cc paddle/fluid/inference/tensorrt/engine.cc +0 -1

未找到文件。
--- 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_);