Quellcode durchsuchen

fix the error of "TypeError: ones_initializer() got multiple values for (#777)

keyword argument 'dtype'".
Yuefeng Zhou vor 8 Jahren
Ursprung
Commit
33bc8b1469
1 geänderte Dateien mit 2 neuen und 2 gelöschten Zeilen
  1. 2 2
      inception/inception/slim/ops.py

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

@@ -91,7 +91,7 @@ def batch_norm(inputs,
     if scale:
       gamma = variables.variable('gamma',
                                  params_shape,
-                                 initializer=tf.ones_initializer,
+                                 initializer=tf.ones_initializer(),
                                  trainable=trainable,
                                  restore=restore)
     # Create moving_mean and moving_variance add them to
@@ -105,7 +105,7 @@ def batch_norm(inputs,
                                      collections=moving_collections)
     moving_variance = variables.variable('moving_variance',
                                          params_shape,
-                                         initializer=tf.ones_initializer,
+                                         initializer=tf.ones_initializer(),
                                          trainable=False,
                                          restore=restore,
                                          collections=moving_collections)