密集层使用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作为后端。