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, ...@@ -76,7 +76,7 @@ def batch_norm(inputs,
""" """
inputs_shape = inputs.get_shape() inputs_shape = inputs.get_shape()
with tf.variable_op_scope([inputs], scope, 'BatchNorm', reuse=reuse): 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:] params_shape = inputs_shape[-1:]
with scopes.arg_scope([variables.variable], restore=restore): with scopes.arg_scope([variables.variable], restore=restore):
# Allocate parameters for the beta and gamma of the normalization. # 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