diff --git a/keras_contrib/backend/tensorflow_backend.py b/keras_contrib/backend/tensorflow_backend.py index 259fefe..0f73034 100644 --- a/keras_contrib/backend/tensorflow_backend.py +++ b/keras_contrib/backend/tensorflow_backend.py @@ -103,7 +103,7 @@ def extract_image_patches(X, ksizes, ssizes, border_mode="same", dim_ordering="t def depth_to_space(input, scale): ''' Uses phase shift algorithm to convert channels/depth for spatial resolution ''' - assert K.image_dim_ordering() == 'tf', 'depth_to_scale backend function can only be used with "tf" dim ' \ + assert K.image_dim_ordering() == 'tf', 'depth_to_space backend function can only be used with "tf" dim ' \ 'ordering when using tensorflow backend' return tf.depth_to_space(input, scale) diff --git a/keras_contrib/backend/theano_backend.py b/keras_contrib/backend/theano_backend.py index 963bc90..314948c 100644 --- a/keras_contrib/backend/theano_backend.py +++ b/keras_contrib/backend/theano_backend.py @@ -121,7 +121,7 @@ def extract_image_patches(X, ksizes, strides, border_mode="valid", dim_ordering= def depth_to_space(input, scale): ''' Uses phase shift algorithm to convert channels/depth for spatial resolution ''' - assert K.image_dim_ordering() == 'th', 'depth_to_scale backend function can only be used with "th" dim ' \ + assert K.image_dim_ordering() == 'th', 'depth_to_space backend function can only be used with "th" dim ' \ 'ordering when using theano backend' b, k, row, col = input.shape