pytorch打印当前学习率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'])
PyTorch学习率打印
本文介绍了如何在PyTorch中打印当前的学习率。通过访问优化器的状态字典,可以轻松获取并显示当前使用的学习率。

8万+

被折叠的 条评论
为什么被折叠?



