(8)Bubbliiiing_yolov4训练代码阅读汇总

本文总结了PyTorch中训练 yolov4 模型的关键知识点,包括参数初始化、模型保存与加载、model.train()与model.eval()的使用、with torch.no_grad()的作用、cudnn.benchmark的优化、随机种子设置、冻结层训练、数据集与数据加载器的运用以及for循环中的enumerate用法。
摘要由CSDN通过智能技术生成

目录

1.pytorch中参数初始化方法

2.pytorch保存与加载模型来测试或继续训练

3.model.train()与model.eval()的用法

4.with torch.no_grad()

5.cudnn.benchmark = True

6.Numpy.random.seed()

7.冻结情况下训练

8.DataSet和DataLoader用法

9.for循环总结(enumerate)


​​​​​​​​​​​​​​

1.pytorch中参数初始化方法

https://www.cnblogs.com/lindaxin/p/8037561.html

yolo_training.py中的weights_init函数采用以下方法对整个网络进行初始化,如图所示:

2.pytorch保存与加载模型来测试或继续训练

https://blog.csdn.net/yanghao201607030101/article/details/110947689

train.py中加载预训练模型进行继续训练,如图所示:

3.model.train()与model.eval()的用法

https://blog.csdn.net/Qy1997/article/details/106455717/

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值