p y t o r c h 打 印 当 前 学 习 率 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'])