ultralytics 8.0.190 add thop>=0.1.1 to requirements.txt (#5162)

Co-authored-by: JohanWesto <Johan.westo@gmail.com>
Co-authored-by: Muhammad Rizwan Munawar <62513924+RizwanMunawar@users.noreply.github.com>
Co-authored-by: StephenBeirlaen <11806615+StephenBeirlaen@users.noreply.github.com>
This commit is contained in:
Glenn Jocher 2023-09-29 21:06:14 +02:00 committed by GitHub
parent 092b58a8cf
commit 9aaa5d5ed0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 28 additions and 11 deletions

View file

@ -627,6 +627,9 @@ class Exporter:
if builder.platform_has_fast_fp16 and self.args.half:
config.set_flag(trt.BuilderFlag.FP16)
del self.model
torch.cuda.empty_cache()
# Write file
with builder.build_engine(network, config) as engine, open(f, 'wb') as t:
# Metadata