pytorch打印当前学习率

$$pytorch打印当前学习率$$

# 打印当前学习率
print(optimizer.state_dict()['param_groups'][0]['lr'])

---

net = major_config.model  # 对应修改模型 net = se_resnet50(num_classes=5,pretrained=True)
optimizer = optim.SGD(net.parameters(), lr=LR, momentum=0.9)   
# 打印当前学习率
print(optimizer.state_dict()['param_groups'][0]['lr'])