浏览代码

Update ops.py

Make sure range create a list, otherwise this won't create a list in Python 3.
Sergio Guadarrama 9 年之前
父节点
当前提交
dd0671b773
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      inception/inception/slim/ops.py

+ 1 - 1
inception/inception/slim/ops.py

@@ -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.