Fix DDP classify train `imgsz` (#2897)

2 years ago · 0d47d11393
parent a2bb42dfe9
commit 0d47d11393
1 changed files with 1 additions and 1 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:
+        if overrides.get('imgsz') is None and cfg['imgsz'] == DEFAULT_CFG.imgsz == 640:
            overrides['imgsz'] = 224
        super().__init__(cfg, overrides, _callbacks)