xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 13:34:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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
Show Stats 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