You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

Lightning 2.0中Trainer.tuner.lr_find报错的代码改写咨询

解决方案:适配PyTorch Lightning 2.0的LR Finder用法

PyTorch Lightning 2.0对调优模块做了重大重构,移除了Trainer实例的tuner属性,原代码的调用方式不再兼容。可以通过以下两种方式改写:

方式1:直接调用Trainer的lr_find方法(推荐)

Lightning 2.0已将lr_find功能直接集成到Trainer类中,无需通过tuner属性调用:

res = trainer.lr_find(
    tft,
    train_dataloaders=train_dataloader,
    val_dataloaders=val_dataloader,
    max_lr=10.0,
    min_lr=1e-6,
)

方式2:使用独立的Tuner类

如果需要使用更多调优功能(如调整批量大小等),可以导入独立的Tuner类并实例化:

from lightning.pytorch.tuner import Tuner

tuner = Tuner(trainer)
res = tuner.lr_find(
    tft,
    train_dataloaders=train_dataloader,
    val_dataloaders=val_dataloader,
    max_lr=10.0,
    min_lr=1e-6,
)

两种方式都能正常执行学习率查找,后续可继续使用res.suggestion()获取推荐的学习率,与旧版本逻辑保持一致。

内容的提问来源于stack exchange,提问作者Roma N

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.19 09:47:02