首页 > 解决方案 > Adam 在 RMSProp 上的问题

问题描述

我对 tensorflow 和 keras 相当陌生。因此,我开始使用 CNN 对 keras bundle 上可用的 CIFAR-10 数据集进行图像分类。我在网上找到了一个非常有用的代码,似乎可以帮助我完成整个过程。运行代码后,我想我会尝试调整它。因此,我将优化器从 RMSProp 更改为 Adam,验证准确性略有提高。但是当我尝试预测图像时,模型几乎总是给出错误的预测。切换回 RMSProp 后,模型开始预测良好。

关于这个问题,我有什么不明白的地方吗?为什么我不能使用 Adam,尽管它给出了更好的准确性?

提前致谢!

https://github.com/spmallick/learnopencv/blob/master/KerasCNN-CIFAR/keras-cnn-cifar10.ipynb

这是我的参考,我将优化器代码从 rmsprop 更改为 Adam 并遇到了上述问题。

标签: pythontensorflowoptimization

解决方案


推荐阅读