Skip to content

Commit 9ada2cb

Browse files
committed
[layers] batch norm for tf 0.12.1
1 parent edd27c7 commit 9ada2cb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorlayer/layers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1924,7 +1924,7 @@ def __init__(
19241924
axis = list(range(len(x_shape) - 1))
19251925

19261926
## 1. beta, gamma
1927-
if tf.__version__ > '0.12' and beta_init == tf.zeros_initializer:
1927+
if tf.__version__ > '0.12.1' and beta_init == tf.zeros_initializer:
19281928
beta_init = beta_init()
19291929
beta = tf.get_variable('beta', shape=params_shape,
19301930
initializer=beta_init,
@@ -1935,7 +1935,7 @@ def __init__(
19351935
)#restore=restore)
19361936

19371937
## 2.
1938-
if tf.__version__ > '0.12':
1938+
if tf.__version__ > '0.12.1':
19391939
moving_mean_init = tf.zeros_initializer()
19401940
else:
19411941
moving_mean_init = tf.zeros_initializer

0 commit comments

Comments
 (0)