ultralytics 8.0.55 unified YOLOv8 model YAMLs (#1475)
This commit is contained in:
parent
701fba4770
commit
25cc07401f
45 changed files with 203 additions and 896 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue