第四十五篇 EfficientNetV1、V2实战
文章目录
摘要🐇
上篇文章我介绍了EfficientNetV1、V2模型的网络结构和DP多卡训练的方法,这篇文章我将和大家一起完成EfficientNet模型的实战。
数据集选用第三篇 制作数据集制作的数据集,模型用EfficientNetV1 B0和EfficientNetV2 S 模型。在这篇文章中,我尽量的简化代码,只保留最基本的逻辑,让每一个初学者能够看明白。通过这篇文章你能学到:
1、如何训练模型?
2、如何推理?
3、如何读取数据集、处理数据集?
4、如何保存权重文件和整个模型文件?
5、如何使用评价指标,如ACC、ReCall等指标评价模型。
6、如何使用matplotlib.pyplot绘制acc和loss曲线图?
7、如何使用余弦退火学习率调整策略调整学习率?
8、如何使用Mixup,CutMix等数据增强?
9、如何使用Pytorch自带的增强方法
10、如何使用argparse模块
11、如何初始化模块
12、使用使用DP多卡
原文地址:https://blog.csdn.net/m0_47867638/article/details/144355650
免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!