This commit is contained in:
Yaojia Wang
2026-02-11 23:40:38 +01:00
parent f1a7bfe6b7
commit ad5ed46b4c
117 changed files with 5741 additions and 7669 deletions

View File

@@ -53,7 +53,7 @@ class TestTrainingConfigSchema:
"""Test default training configuration."""
config = TrainingConfig()
assert config.model_name == "yolo11n.pt"
assert config.model_name == "yolo26s.pt"
assert config.epochs == 100
assert config.batch_size == 16
assert config.image_size == 640
@@ -63,7 +63,7 @@ class TestTrainingConfigSchema:
def test_custom_config(self):
"""Test custom training configuration."""
config = TrainingConfig(
model_name="yolo11s.pt",
model_name="yolo26s.pt",
epochs=50,
batch_size=8,
image_size=416,
@@ -71,7 +71,7 @@ class TestTrainingConfigSchema:
device="cpu",
)
assert config.model_name == "yolo11s.pt"
assert config.model_name == "yolo26s.pt"
assert config.epochs == 50
assert config.batch_size == 8
@@ -136,7 +136,7 @@ class TestTrainingTaskModel:
def test_task_with_config(self):
"""Test task with configuration."""
config = {
"model_name": "yolo11n.pt",
"model_name": "yolo26s.pt",
"epochs": 100,
}
task = TrainingTask(