Skip to content

Commit 1f2c85b

Browse files
committed
[layers] batchnorm layer for tf11 and tf12
1 parent 7a877ac commit 1f2c85b

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

tensorlayer/layers.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1731,10 +1731,16 @@ def _get_variable(name,
17311731
params_shape,
17321732
initializer=tf.zeros_initializer,
17331733
trainable=False)
1734-
moving_variance = _get_variable('moving_variance',
1735-
params_shape,
1736-
initializer=tf.ones_initializer(),
1737-
trainable=False)
1734+
try: # TF12
1735+
moving_variance = _get_variable('moving_variance',
1736+
params_shape,
1737+
initializer=tf.ones_initializer(),
1738+
trainable=False)
1739+
except: # TF11
1740+
moving_variance = _get_variable('moving_variance',
1741+
params_shape,
1742+
initializer=tf.ones_initializer,
1743+
trainable=False)
17381744

17391745
# These ops will only be preformed when training.
17401746
mean, variance = tf.nn.moments(self.inputs, axis)

0 commit comments

Comments
 (0)