在没提出ResNet模型之前,传统深度模型随着模型的加深 准确率降低 误差更大

ResNet提出了残差网络,可以轻松搭建超过1000层的神经网络,使模型越深准确率有所提升。

# CIFAR-10数据集默认mean 和std
vision.Normalize([0.4914, 0.4822, 0.4465], [0.2023, 0.1994, 0.2010])

ResNet50预训练模型默认输出是1000分类 需要指定最后一层全连接输出为10

 

Logo

鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。

更多推荐