ultralytics 8.0.174 Tuner plots and improvements (#4799)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-09-10 03:27:23 +02:00 · 2023-09-10 03:27:23 +02:00 · 16ce193d6e
commit 16ce193d6e
parent dfe6dfb1d2
12 changed files with 248 additions and 88 deletions
--- a/ultralytics/hub/session.py
+++ b/ultralytics/hub/session.py
@ -117,8 +117,7 @@ class HUBTrainingSession:

            if data['status'] == 'new':  # new model to start training
                self.train_args = {
-                    # TODO: deprecate 'batch_size' key for 'batch' in 3Q23
-                    'batch': data['batch' if ('batch' in data) else 'batch_size'],
+                    'batch': data['batch'],
                    'epochs': data['epochs'],
                    'imgsz': data['imgsz'],
                    'patience': data['patience'],