wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-06-27 16:10:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix tests

Browse Source
This commit is contained in:
Somshubra Majumdar
2017-02-27 22:54:52 -06:00
parent 4aa331d548
commit 1986348465
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/keras_contrib/layers/test_convolutional.py
+5
View File
@@ -86,6 +86,11 @@ def test_sub_pixel_upscaling():
for scale_factor in [2, 3, 4]:
input_data = np.random.random((nb_samples, 4 * (scale_factor ** 2), nb_row, nb_col))
if K.backend() == 'theano':
K.set_image_dim_ordering('th')
else:
K.set_image_dim_ordering('tf')
if K.image_dim_ordering() == 'tf':
input_data = input_data.transpose((0, 2, 3, 1))