- 博客(2)
- 收藏
- 关注
原创 第一次作业总结:树叶分类
回头看自己的第一次作业,以及李牧老师的课程学习记录。想起来一句话: 及时当勉励,岁月不待人。1.第一次作业的处理过程的心得。
2024-05-19 17:35:35
78
1
原创 第一次作业:kaggle比赛:Classify Leaves图片树叶分类。
因为下面的层学习的是一些底层的细节,低层次的特征更加的通用,所以我们固定下面的层,把底层copy过来,参数拿过来用即可。前面我们已知总有有176个类,所以最后做预测的时候只需要输出176,我们将num_classes=176,resnet50模型的输出是1000,所以我们加了一个全连接层。这三个函数的作用是先将梯度归零(optimizer.zero_grad()),然后反向传播计算得到每个参数的梯度值(loss.backward()),最后通过梯度下降执行一步参数更新(optimizer.step())
2024-05-19 17:25:09
718
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人