- 博客(3)
- 收藏
- 关注
原创 ResNet18训练cifar-10数据集
可以看见模型已经收敛,但是测试的精度只有85%,模型陷入了局部最优解。后续看看能不能进行优化。在Dataloader函数让shuffle=True进行洗牌。每隔2个epoch就将学习乘以0.9。在首先进行了数据增广和数据标准化。
2023-06-02 11:04:24 296
原创 VggNet训练cifar-10数据集
这里可以看见每次学习率下降时,测试精度会有小幅度提升,调小学习率可以改善过拟合,可以看见在25轮后学习下降了,26轮后的测试精度又开始小幅度上升,大概可以上升到0.77。一开始实现了vgg网络使用Adam优化器设置的学习率0.1,测试的loss毫无变化,原因是学习率learning rate太搞了后来设置成0.005,并且使用xavier_normal初始化卷积层和全连接层的参数,loss才开始慢慢下降,但是精度达到0.71就开始overfitting了。之后改变优化器使用SDG。
2023-05-30 16:36:11 420 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人