From 07914f0bbfc1c33e97e2a0ea1df5205594208f50 Mon Sep 17 00:00:00 2001 From: TianchengHu <3139658+tiancheng2000@users.noreply.github.com> Date: Sun, 10 May 2020 01:27:33 +0800 Subject: [PATCH] only compute if necessary --- tensorlayer/layers/normalization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorlayer/layers/normalization.py b/tensorlayer/layers/normalization.py index 39db51159..dd7cc2189 100644 --- a/tensorlayer/layers/normalization.py +++ b/tensorlayer/layers/normalization.py @@ -291,9 +291,9 @@ def forward(self, inputs): if self.axes is None: self.axes = [i for i in range(len(inputs.shape)) if i != self.channel_axis] - mean, var = tf.nn.moments(inputs, self.axes, keepdims=False) if self.is_train: # update moving_mean and moving_var + mean, var = tf.nn.moments(inputs, self.axes, keepdims=False) self.moving_mean = moving_averages.assign_moving_average( self.moving_mean, mean, self.decay, zero_debias=False )