如何修复 python pytorch 库版本问题？

Question

警告（来自警告模块）： 文件“C:\Users\mesut\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\optim\lr_scheduler.py”，第 134 行 “https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate”，用户警告） 用户警告：在optimizer.step() 之前检测到lr_scheduler.step() 的调用。在 PyTorch 1.1.0 及更高版本中，您应该以相反的顺序调用它们：optimizer.step() 在 lr_scheduler.step() 之前。不这样做将导致 PyTorch 跳过学习率计划的第一个值。在https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate查看更多详情

Answer 1

我认为你没有版本问题。

警告通常发生在您在更新实际模型权重之前更改学习率时。

在您似乎拥有的最新版本的 Pytorch 中，代码会输出警告，因为在模型上进行单一优化之前更改学习率是没有意义的。而是首先以不同的方式初始化您的学习率。

因此，警告说 Pytorch 跳过 lr 调度程序的第一个值，只是首先进行优化。但是，这是多余的，不是一个好的代码实现，因此会发出警告。

因此，我建议你在第一次运行时摆脱这个错误

optimizer.step()
lr_scheduler.step()

而不是您似乎拥有的其他方式。

萨塔克·耆那教