Commit c74897b0 authored by Sergio Guadarrama's avatar Sergio Guadarrama
Browse files

Merge pull request #42 from tensorflow/sguada-patch-1

Update ops.py
parents 044a5f5b dd0671b7
......@@ -76,7 +76,7 @@ def batch_norm(inputs,
"""
inputs_shape = inputs.get_shape()
with tf.variable_op_scope([inputs], scope, 'BatchNorm', reuse=reuse):
axis = range(len(inputs_shape) - 1)
axis = list(range(len(inputs_shape) - 1))
params_shape = inputs_shape[-1:]
with scopes.arg_scope([variables.variable], restore=restore):
# Allocate parameters for the beta and gamma of the normalization.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment