From 56f3c85b87c2fbe75d9d791b7eb924bd455776d3 Mon Sep 17 00:00:00 2001 From: Seonghyeon Nam Date: Fri, 16 Sep 2016 10:09:02 +0900 Subject: [PATCH] Fix ValueError(ndim of gamma and beta) of batch normalization when using Theano (#3740) * Fix ndim mismatch error when using theano * Change keras backend call --- keras/layers/normalization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/keras/layers/normalization.py b/keras/layers/normalization.py index 47b92f0cf..6e48ff24c 100644 --- a/keras/layers/normalization.py +++ b/keras/layers/normalization.py @@ -140,7 +140,7 @@ class BatchNormalization(Layer): self.updates = [K.moving_average_update(self.running_mean, mean, self.momentum), K.moving_average_update(self.running_std, std, self.momentum)] - if sorted(reduction_axes) == range(K.ndim(x))[:-1]: + if K.backend() == 'tensorflow' and sorted(reduction_axes) == range(K.ndim(x))[:-1]: x_normed_running = K.batch_normalization( x, self.running_mean, self.running_std, self.beta, self.gamma,