由于种种原因, 需要自己写一个lr_scheduler,
目前主流的方法是从optimizer中获取,
代码如下:
optimizer.state_dict()['param_groups'][0]['lr']
PyTorch如何获取当前的学习率Learning Rate(lr)
猜你喜欢
转载自blog.csdn.net/ftimes/article/details/120975402
今日推荐
周排行