Преглед на файлове

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()
   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.