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 tf dim ordering test

Browse Source
This commit is contained in:
Michael Oliver
2017-02-22 18:23:39 -08:00
parent 5b273df036
commit d8ca85f445
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/keras_contrib/layers/test_convolutional.py
+1 -1
View File
@@ -121,7 +121,7 @@ def test_cosineconvolution_2d():
elif dim_ordering == 'tf':
X = np.random.randn(1, 5, 5, 3)
input_dim = (5, 5, 3)
W0 = X[:, ::-1, ::-1, :]
W0 = X[0, ::-1, ::-1, :, None]
model = Sequential()
model.add(convolutional.CosineConvolution2D(1, 5, 5, bias=True, input_shape=input_dim))