本例提取了植物幼苗数据集中的部分数据做数据集,数据集共有12种类别,模型使用最经典的resnet50,演示如何实现混合精度训练以及如何使用DDP的方式实现多卡并行训练。
通过本文你和学到:
1、如何使用混合精度训练?
2、如何制作ImageNet数据集?
3、如何使用DDP方式的进行多卡训练?
4、如何使用Mixup数据增强。
5、如何进行多卡BN同步?
6、如何使用余弦退火调整学习率?
7、如何使用classification_report实现对模型的评价。
8、预测的两种写法。
原文链接:https://blog.csdn.net/hhhhhhhhhhwwwwwwwwww/article/details/124196817
2022-04-15 18:13:12
930.99MB
图像分类