未验证 提交 80092576 编写于 作者: W Wangzheee 提交者: GitHub

fix_trt_int8 (#34704)

上级 7afd31bb
...@@ -89,7 +89,6 @@ void TensorRTEngine::FreezeNetwork() { ...@@ -89,7 +89,6 @@ void TensorRTEngine::FreezeNetwork() {
if (enable_int8) { if (enable_int8) {
infer_builder_config_->setFlag(nvinfer1::BuilderFlag::kFP16); infer_builder_config_->setFlag(nvinfer1::BuilderFlag::kFP16);
infer_builder_config_->setFlag(nvinfer1::BuilderFlag::kINT8); infer_builder_config_->setFlag(nvinfer1::BuilderFlag::kINT8);
infer_builder_config_->setFlag(nvinfer1::BuilderFlag::kSTRICT_TYPES);
if (calibrator_) { if (calibrator_) {
infer_builder_config_->setInt8Calibrator(calibrator_); infer_builder_config_->setInt8Calibrator(calibrator_);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册