Introduce Keras TF atrous bug

israelg99 · Feb 26, 2017 · b9a28ad · b9a28ad
1 parent 6fc53ec
commit b9a28ad
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/keras_tf_atrous_bug.py b/keras_tf_atrous_bug.py
@@ -0,0 +1,10 @@
+import numpy as np
+from keras.models import Model
+from keras.layers import Input, AtrousConvolution1D
+
+inp = Input((100, 1))
+M = AtrousConvolution1D(1, 2, atrous_rate=25, border_mode='same')(inp)
+M = Model(inp, M)
+M.compile('sgd', 'mse')
+
+M.train_on_batch(np.random.rand(1, 100, 1), np.random.rand(1, 100, 1))