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训练深度学习小技巧收集的主要内容,如果未能解决你的问题,请参考以下文章

六PyTorch进阶训练技巧

Pytorch:实战指南

[深度学习][pytorch]pytorch实现一个简单得线性回归模型并训练

PyTorch学习笔记:PyTorch进阶训练技巧

PyTorch经验指南:技巧与陷阱

神经网络和深度学习-用pytorch实现线性回归