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,

最低0.47元/天 解锁文章
1297

被折叠的 条评论
为什么被折叠?



