Update generate_ddp_file for improved overrides (#2909)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-05-31 01:41:44 +08:00
parent facb7861cf
commit 305cde69d0
3 changed files with 9 additions and 8 deletions
--- a/ultralytics/yolo/v8/classify/train.py
+++ b/ultralytics/yolo/v8/classify/train.py
@ -19,7 +19,7 @@ class ClassificationTrainer(BaseTrainer):
        if overrides is None:
            overrides = {}
        overrides['task'] = 'classify'
-        if overrides.get('imgsz') is None and cfg['imgsz'] == DEFAULT_CFG.imgsz == 640:
+        if overrides.get('imgsz') is None:
            overrides['imgsz'] = 224
        super().__init__(cfg, overrides, _callbacks)