ultralytics 8.0.175 StreamLoader wait for missing frames (#4814)

This commit is contained in:
Glenn Jocher 2023-09-10 23:59:43 +02:00 committed by GitHub
parent dd0782bd8d
commit 3c88bebc95
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
11 changed files with 197 additions and 49 deletions

View file

@ -117,7 +117,7 @@ class HUBTrainingSession:
if data['status'] == 'new': # new model to start training
self.train_args = {
# TODO deprecate before 8.1.0
# TODO deprecate 'batch_size' argument in favor of 'batch'
'batch': data['batch' if 'batch' in data else 'batch_size'],
'epochs': data['epochs'],
'imgsz': data['imgsz'],