Fix DDP classify train imgsz (#2897)

2023-05-29 19:23:19 +02:00 · 2023-05-29 19:23:19 +02:00 · 0d47d11393
commit 0d47d11393
parent a2bb42dfe9
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)