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:
parent
092b58a8cf
commit
9aaa5d5ed0
7 changed files with 28 additions and 11 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue