Ultralytics YOLO超参数调整指南(二)
Ultralytics YOLO超参数列表和取值区间请阅读:Ultralytics YOLO超参数调整指南(一)
自定义搜索空间示例
下面介绍如何定义搜索空间并使用 model.tune()
方法来利用 Tuner
在 COCO8 上使用 AdamW 优化器对 YOLO11n 的超参数进行 30 个轮次的调整,并跳过绘图、检查点和验证(最后一个轮次除外),以加快调整速度。
from ultralytics import YOLO# Initialize the YOLO model
model = YOLO("yolo11n.pt")# Define search space
search_space = {"lr0": (1e-5, 1e-1),"degrees": (0.0, 45.0),
}# Tune hyperparameters on COCO8 for 30 epochs
model.tune(data="coco8.yaml",epochs=30,iterations=300,optimizer="AdamW",space=search_space,plots=False