>- **🍨 本文为[🔗365天深度学习训练营](https://mp.weixin.qq.com/s/0dvHCaOoFnW8SCp3JpzKxg) 中的学习记录博客**
>- **🍖 原作者:[K同学啊](https://mtyjkh.blog.csdn.net/)**
一、 前期准备
运行结果
二、调用官方的VGG-16模型
运行结果
三、 训练模型
运行结果
测试集准确率较低,约为20%,以下对优化器和初始学习率进行调整
1.将初始学习率从0.0001调整为0.001
结果为30%左右
2.将初始学习率从0.0001调整为0.01;并将优化器修改为Adam
运行结果为
3.将初始学习率从0.0001调整为0.001;并将优化器修改为Adam,训练次数改为50
结果
四、 结果可视化
出现了不显示图片预测结果的问题,对代码进行调整
结果为
五、总结
最终准确率也只达到了50%左右,没有达到预期,初始学习率的设置和优化器的使用,梯度下降的速率,都会影响最终的结果,进行下一步改进可以采取改变VGG框架的方式。