使用 Keras 和 Theano 的密集层

Question

我正在开始一个简单的 keras 教程。我正在使用的完整代码在本网站的末尾：

https://elitedatascience.com/keras-tutorial-deep-learning-in-python

我收到此错误：

ValueError: ('The specified size contains a dimension with value <= 0', (-768, 128))

对于 Dense 出现的行：

model.add(Dense(128, activation='relu'))

和

model.add(Dense(10, activation='softmax'))

我正在使用这个 json 文件进行后端配置：

{
"image_dim_ordering": "tf",
"epsilon": 1e-07, 
"floatx": "float32", 
"backend": "theano"
}

知道问题出在哪里吗？我正在使用 python2.7 并且 MNIST 数据库似乎是正确的，因为我能够绘制图像并且还可以毫无问题地查看标签。

最好的，

天使。

Answer 1

尝试将 "image_dim_ordering": "tf" 设置为 "image_dim_ordering": "th"，因为您使用 Theano 作为后端。