ultralytics 8.1.38 fix deprecated Ray Tune .is_session_enabled() (#9432)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-03-31 00:13:05 +01:00 committed by GitHub
parent 007f778b72
commit ea80b14d72
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 11 additions and 8 deletions

View file

@ -81,6 +81,8 @@ class BaseDataset(Dataset):
if self.rect:
assert self.batch_size is not None
self.set_rectangle()
if isinstance(cache, str):
cache = cache.lower()
# Buffer thread for mosaic images
self.buffer = [] # buffer size = batch size