Add timing cache to accelerate consequent .engine export (#13386)

* fix: typos

* feat: enable timing cache for engine export

* Auto-format by https://ultralytics.com/actions

---------

Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
imyhxyandUltralyticsAssistant authored and GitHub committed 2024-11-08 23:24:30 +01:00
1 parent 3760e0ed93
commit 1435a8eed6
2 files changed
+22 -6

No files matched your search

+3 -3
View File
@@ -717,10 +717,10 @@ def main(opt, callbacks=Callbacks()):
"perspective": (True, 0.0, 0.001), # image perspective (+/- fraction), range 0-0.001
"flipud": (True, 0.0, 1.0), # image flip up-down (probability)
"fliplr": (True, 0.0, 1.0), # image flip left-right (probability)
"mosaic": (True, 0.0, 1.0), # image mixup (probability)
"mosaic": (True, 0.0, 1.0), # image mosaic (probability)
"mixup": (True, 0.0, 1.0), # image mixup (probability)
"copy_paste": (True, 0.0, 1.0),
} # segment copy-paste (probability)
"copy_paste": (True, 0.0, 1.0), # segment copy-paste (probability)
}
# GA configs
pop_size = 50