【python学习】简述pytorch中optimizer.step()的用法

参考:

https://www.jb51.net/article/213737.htmhttps://www.jb51.net/article/213737.htmpytorch里面的Optimizer和optimizer.step()用法_wang xiang的博客-CSDN博客_optimizer.step()用法参考官方文档链接当我们想指定每一层的学习率时:optim.SGD([{‘params’: model.base.parameters()},{‘params’: model.classifier.parameters(), ‘lr’: 1e-3}], lr=1e-2, momentum=0.9)这意味着model.base的参数将会使用1e-2的学习率,model.classifi...https://blog.csdn.net/qq_40178291/article/details/99963586

你可能感兴趣的:(Pytorch框架学习,pytorch,人工智能,python)