Update YouTube URL https://youtu.be/LNwODJXcvt4 (#4808)

2023-09-10 04:58:24 +02:00 · 2023-09-10 04:58:24 +02:00 · dbc53f6741
commit dbc53f6741
parent 16ce193d6e
12 changed files with 26 additions and 25 deletions
--- a/ultralytics/hub/session.py
+++ b/ultralytics/hub/session.py
@ -117,7 +117,8 @@ class HUBTrainingSession:

            if data['status'] == 'new':  # new model to start training
                self.train_args = {
-                    'batch': data['batch'],
+                    # TODO deprecate before 8.1.0
+                    'batch': data['batch' if 'batch' in data else 'batch_size'],
                    'epochs': data['epochs'],
                    'imgsz': data['imgsz'],
                    'patience': data['patience'],