python - AttributeError:模块“torch.optim.lr_scheduler”没有属性“LinearLR”
问题描述
我正在尝试使用 Pytorch 训练我自己的对象检测模型。但我总是得到这个错误。我试图更改手电筒版本,但这没有帮助。
我的包:torchvision-0.11.1 和 torch-1.10.0
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
<ipython-input-10-9e52b782b448> in <module>()
4 for epoch in range(num_epochs):
5 # training for one epoch
----> 6 train_one_epoch(model, optimizer, data_loader, device, epoch, print_freq=10)
7 # update the learning rate
8 lr_scheduler.step()
/content/engine.py in train_one_epoch(model, optimizer, data_loader, device, epoch, print_freq)
21 warmup_iters = min(1000, len(data_loader) - 1)
22
---> 23 lr_scheduler = torch.optim.lr_scheduler.LinearLR(
24 optimizer, start_factor=warmup_factor, total_iters=warmup_iters
25 )
AttributeError: module 'torch.optim.lr_scheduler' has no attribute 'LinearLR'
解决方案
LinearLR
调度程序是最近才引入的(v1.10.0)。请确保您的 pytorch 版本是最新的,然后重试。
推荐阅读
- google-drive-api - 无论如何,是否可以使用 Google Drive API 列出哪些文件可以与任何有链接的人共享或可以公开搜索?
- php - 使用 Apache 和 Kohana 创建 PHP 服务器
- sas - 使用数据步骤中的 put 语句写入 SAS EG 中的默认 HTML 输出
- reactjs - 将选定的值保留在 React 的一组单选按钮中
- html - 由于全宽行,引导 4 网格空白
- javascript - Sequelize 模型关联不会创建新列
- csv - Wix:在 UI 序列中读取文本文件
- php - php @chdir 文件未找到
- django - 如何在石墨烯解析器中调用 django 模型函数?
- pandas - 如何在熊猫中使用 crosstab .plot() 绘制图表中的轴顺序?