pytorch深度学习,迭代优化,迁移学习步骤,引用现有模型pth,并且修改模型训练过程,一看就懂系列

本文介绍了使用PyTorch进行深度学习的迁移学习步骤,包括加载模型实例、冻住部分卷积层、修改全连接层参数。同时探讨了迭代优化的重要性,分析了在设计CNN模型时如何选择合适的卷积层数量,以及在训练过程中如何通过调整epoch和batch大小来防止过拟合,强调保存最佳模型的重要性。
摘要由CSDN通过智能技术生成

迁移学习
1,首先加载模型实例
2,冻住卷积层(或者其中一部分),直修改全链接层
3,拿到最后一层全链接层的输入层数值
4,修改最后一层的全链接层参数矩阵
迭代优化
在整个训练过程中,存在两个地方需要迭代优化的选择
1,是在设计卷积神经网络的时候,需要优化的地方,不一定是卷积层数量越多,loss值就越小,它有一个瓶颈例如在VGG网络模型中,但是最后发现是因为我们设计的卷基层的卷积核某个层可能不是特别好,为了解决这个问题,出现了RESNET,残余差错的网络模型,它可以丢掉不好的隐藏层,
2,是在训练的过程中,按照epoch来规定训练多少次,batch规定一次训练多少个样本,例如一个bh训练64张图片,或者128张图片,我们需要在训练的过程中观察并保存loss值,保存最好的那一次的epoch训练得到的模型结果,如果结果好的化就要保存,防止过拟合的出现

已下架不支持订阅

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

鼾声鼾语

感谢您的支持鼓励!

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值