ソースを参照

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

Update ops.py
Sergio Guadarrama 9 年 前
コミット
c74897b05c
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()
   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.