We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7a877ac commit 1f2c85bCopy full SHA for 1f2c85b
tensorlayer/layers.py
@@ -1731,10 +1731,16 @@ def _get_variable(name,
1731
params_shape,
1732
initializer=tf.zeros_initializer,
1733
trainable=False)
1734
- moving_variance = _get_variable('moving_variance',
1735
- params_shape,
1736
- initializer=tf.ones_initializer(),
1737
- trainable=False)
+ try: # TF12
+ moving_variance = _get_variable('moving_variance',
+ params_shape,
+ initializer=tf.ones_initializer(),
1738
+ trainable=False)
1739
+ except: # TF11
1740
1741
1742
+ initializer=tf.ones_initializer,
1743
1744
1745
# These ops will only be preformed when training.
1746
mean, variance = tf.nn.moments(self.inputs, axis)
0 commit comments