PyTorch如何获取当前的学习率Learning Rate(lr)

由于种种原因, 需要自己写一个lr_scheduler,
目前主流的方法是从optimizer中获取,
代码如下:

optimizer.state_dict()['param_groups'][0]['lr']

你可能感兴趣的:(pytorch,深度学习,人工智能)