ultralytics 8.0.55 unified YOLOv8 model YAMLs (#1475)

2023-03-20 13:54:20 +01:00
parent 701fba4770
commit 25cc07401f
45 changed files with 203 additions and 896 deletions
--- a/tests/test_engine.py
+++ b/tests/test_engine.py
@ -74,16 +74,9 @@ def test_segment():


 def test_classify():
-    overrides = {
-        'data': 'imagenet10',
-        'model': 'yolov8n-cls.yaml',
-        'imgsz': 32,
-        'epochs': 1,
-        'batch': 64,
-        'save': False}
+    overrides = {'data': 'imagenet10', 'model': 'yolov8n-cls.yaml', 'imgsz': 32, 'epochs': 1, 'save': False}
    CFG.data = 'imagenet10'
    CFG.imgsz = 32
-    CFG.batch = 64
    # YOLO(CFG_SEG).train(**overrides)  # works

    # Trainer