- 博客(9)
- 收藏
- 关注
RSS订阅原创 图像识别实战代码分析
引入一个新的概念，迁移学习，将公开模型已经训练好的参数作为待训练模型参数的初始化。假如，我们使用除输出层之外的所有层的参数，就将这些参数“冻住”，在模型训练的过程中，不更新这些我参数。引包：torchvision，包含transforms、model（在这指的就是网络的模型）、datasets（对数据的处理）torch的数据为C*H*W，分别占「0，1，2」号位，PIL所需要的数据为H*W*C，transpose为调整参数的位置。5、加载训练好的模型，存储训练的权重参数、偏执、训练的准确率、损失。
2024-03-31 14:42:42
1286
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人