HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

TensorRT FP16 export with dynamic=True (#7870)

Browse source
This commit is contained in:
Yonghye Kwon 2024-01-29 21:54:06 +09:00 • committed by GitHub
parent 51fb3d607e
commit 9bdd784bcc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/engine/exporter.py
View file

@ -227,7 +227,7 @@ class Exporter:
y = None y = None
for _ in range(2): for _ in range(2):
y = model(im) # dry runs y = model(im) # dry runs
if self.args.half and (engine or onnx) and self.device.type != "cpu": if self.args.half and onnx and self.device.type != "cpu":
im, model = im.half(), model.half() # to FP16 im, model = im.half(), model.half() # to FP16
# Filter warnings # Filter warnings