From c02be31412f961809847fac128be44c1c9a19e2b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 4 Jan 2022 19:34:34 -0800 Subject: [PATCH] 7 to 8 fix --- export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/export.py b/export.py index 17b994df82a4..c56a0a99a635 100644 --- a/export.py +++ b/export.py @@ -179,7 +179,7 @@ def export_engine(model, im, file, train, half, simplify, workspace=4, verbose=F export_onnx(model, im, file, 12, train, False, simplify) # opset 12 model.model[-1].anchor_grid = grid else: # TensorRT >= 8 - check_version(trt.__version__, '7.0.0', hard=True) # require tensorrt>=8.0.0 + check_version(trt.__version__, '8.0.0', hard=True) # require tensorrt>=8.0.0 export_onnx(model, im, file, 13, train, False, simplify) # opset 13 onnx = file.with_suffix('.onnx') assert onnx.exists(), f'failed to export ONNX file: {onnx}'