- 博客(2)
- 收藏
- 关注
RSS订阅原创 Pytorch：lr_schedule恢复训练的注意事项
在训练过程中我们一般会使用pytorch已有的学习率调整策略，如：import torchimport torch.optim as optimfrom torchvision.models.resnet import resnet50net = resnet50(num_classes=1000)optimizer = optim.Adam(net.parameters(), lr...
2019-01-03 14:40:25
12724
12
原创 Pytorch采用不同的学习率训练网络
Pytorch采用不同的学习率训练网络为了fine-tuning网络，经常会让有预训练参数的层的学习率比较小，而新的层的学习率比较大。然后看了网上很多的教程，都是这样去定义参数组的：# eg.import torchfrom torchvision.models.vgg import vgg16_bnfrom torch import optimnum_classes = 2lr ...
2018-12-12 14:02:05
788
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人