mirror of
https://github.com/THU-MIG/yolov10.git
synced 2025-05-23 13:34:23 +08:00
5b3c4cfc0e
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
27 lines
1020 B
Python
27 lines
1020 B
Python
# Ultralytics YOLO 🚀, AGPL-3.0 license
|
|
|
|
import pytest
|
|
|
|
from ultralytics import YOLO
|
|
from ultralytics.utils import SETTINGS, checks
|
|
|
|
|
|
@pytest.mark.skipif(not checks.check_requirements('ray', install=False), reason='RayTune not installed')
|
|
def test_model_ray_tune():
|
|
"""Tune YOLO model with Ray optimization library."""
|
|
YOLO('yolov8n-cls.yaml').tune(use_ray=True,
|
|
data='imagenet10',
|
|
grace_period=1,
|
|
iterations=1,
|
|
imgsz=32,
|
|
epochs=1,
|
|
plots=False,
|
|
device='cpu')
|
|
|
|
|
|
@pytest.mark.skipif(not checks.check_requirements('mlflow', install=False), reason='MLflow not installed')
|
|
def test_mlflow():
|
|
"""Test training with MLflow tracking enabled."""
|
|
SETTINGS['mlflow'] = True
|
|
YOLO('yolov8n-cls.yaml').train(data='imagenet10', imgsz=32, epochs=3, plots=False, device='cpu')
|