cifar10数据集上的CNN模型似乎未受过训练

时间:2018-10-26 09:43:48

标签: python tensorflow conv-neural-network

我只想看一看有关cnn的简单教程,我想训练由5个滤镜(3个转换层和2个密集层)组成的网络 其次是max_pool层和辍学层。

我的问题是我的网络似乎根本没有经过培训,我找不到问题出在哪里。...

我将所有代码(train.py test.py model.py和main.py)放在github上的以下链接中。

https://github.com/mmecoco/tensorflow_cnn_cifar10

帮帮我...

1 个答案:

答案 0 :(得分:0)

我刚刚发现了我的问题,这是我最后一个fc层引起的,问题是我试图在输出上应用softmax激活,如果我使用softmax交叉熵损失函数,则损失的值将在-inf或inf一侧。 因此,网络停止了学习过程,这就是为什么我的培训失败了。