pytorch训练深度学习小技巧收集
Posted KPer_Yang
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了pytorch训练深度学习小技巧收集相关的知识,希望对你有一定的参考价值。
1、对不同的网络层配置不同的学习率
import torch
optimizer = torch.optim.Adam([
dict(params=model.conv1.parameters(), weight_decay=5e-4),
dict(params=model.conv2.parameters(), weight_decay=0)
], lr=args.lr) # Only perform weight-decay on first convolution.
以上是关于pytorch训练深度学习小技巧收集的主要内容,如果未能解决你的问题,请参考以下文章