Skip to content

Commit 24936bb

Browse files
committed
fixed batch norm
1 parent 3dec543 commit 24936bb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorlayer/layers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1915,7 +1915,7 @@ def mean_var_with_update():
19151915
# lambda: (moving_mean, moving_variance)) # when inferencing, (x-0)/1
19161916
#
19171917
# self.outputs = act( tf.nn.batch_normalization(self.inputs, mean, var, beta, gamma, epsilon) )
1918-
if not is_train:
1918+
if is_train:
19191919
mean, var = mean_var_with_update()
19201920
self.outputs = act( tf.nn.batch_normalization(self.inputs, mean, var, beta, gamma, epsilon) )
19211921
else:

0 commit comments

Comments
 (0)