yolov5查看学习率
时间: 2023-10-16 22:08:08 浏览: 99
yolov5s模型,yolov5学习
你可以通过以下代码查看 YOLOv5 模型的当前学习率:
```python
from models.yolo import Model, is_parallel
model = Model()
optimizer = torch.optim.Adam(model.parameters(), lr=0.001)
print(optimizer.param_groups[0]['lr'])
```
这里假设模型定义在 `models/yolo.py` 文件中,Adam 优化器使用学习率为 0.001。`optimizer.param_groups['lr']` 可以获取当前学习率的值。
阅读全文