lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Keras is TF first. Fix TH first example (#3914)

Browse Source 
* Keras is TF first. Fix TH first example

* Use K.set_image_dim_ordering('th')
This commit is contained in:
Eder Santana 2016-09-29 13:57:08 -04:00 committed by François Chollet
parent 51c85dd8d6
commit 3bf8964355
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/variational_autoencoder_deconv.py
View File

@ -12,6 +12,8 @@ from keras import backend as K
from keras import objectives
from keras.datasets import mnist
K.set_image_dim_ordering('th') # this is a Theano oriented example
# input image dimensions
img_rows, img_cols, img_chns = 28, 28, 1
# number of convolutional filters to use