module 'torch.optim.lr_scheduler' has no attribute 'LRScheduler'

Question

AttributeError: module 'torch.optim.lr_scheduler' has no attribute 'LRScheduler'
用的pytorch是1.13.1+cu116的

HELP! 这个该怎么解决呢？升级torch吗？还是怎么解决呢？

（torch版本是1.12.0的时候没有出现这个问题，但是更新到1.13.1就出现这样的问题了 捣鼓好久了都没有解决

Answer 1

让阿豪来帮你解答，本回答参考chatgpt3.5编写提供，如果还有疑问可以评论或留言

根据报错信息，可以看出是在导入`LRScheduler`时发生了错误。在最新的PyTorch版本（1.13.1），`LRScheduler`已经被弃用，因此无法引用。 解决方法是使用新的学习率调度器类，例如`StepLR`、`MultiStepLR`、`CosineAnnealingLR`等，这些类都在`torch.optim.lr_scheduler`模块中。根据具体需求，选择其中一种调度器，并根据官方文档进行使用。 举个例子，使用`StepLR`调度器，每次训练完一个epoch后将学习率乘以`gamma`：

import torch.optim as optim
import torch.optim.lr_scheduler as lr_scheduler
optimizer = optim.SGD(model.parameters(), lr=0.1)
scheduler = lr_scheduler.StepLR(optimizer, step_size=1, gamma=0.1)
for epoch in range(10):
    # train model
    train_loss = ...
    
    # update learning rate
    scheduler.step()

在每个epoch结束后，调用scheduler.step()即可更新学习率。其中，step_size表示多少个epoch更新一次学习率，gamma表示乘以的系数。 总结一下，解决AttributeError: module 'torch.optim.lr_scheduler' has no attribute 'LRScheduler'错误的方法是： 1. 停止使用LRScheduler调度器； 2. 使用其他调度器类，例如StepLR等； 3. 根据需求设置调度器参数，参考官方文档； 4. 在每个epoch结束后手动调用scheduler.step()更新学习率。 希望以上解答能够帮到你。