python - PyTorch-如何在训练中获得学习率？

在训练期间，我想知道学习率的价值。
我该怎么办？
我的代码是这样的：

my_optimizer = torch.optim.SGD(my_model.parameters(),
                               lr=0.001,
                               momentum=0.99,
                               weight_decay=2e-3)

谢谢您。

最佳答案

对于给定的示例中只有一个参数组，您可以使用此函数并在训练期间调用它以获得当前学习速率：

def get_lr(optimizer):
    for param_group in optimizer.param_groups:
        return param_group['lr']

关于python - PyTorch-如何在训练中获得学习率？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/52660985/