pytorch打印当前学习率

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'])
上一篇:如何只训练网络中的全连接层


下一篇:梯度累加(Gradient Accumulation)