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

Update ops.py

Make sure range create a list, otherwise this won't create a list in Python 3.
parent 044a5f5b
......@@ -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