Selaa lähdekoodia

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

keyword argument 'dtype'".
Yuefeng Zhou 8 vuotta sitten
vanhempi
commit
33bc8b1469
1 muutettua tiedostoa jossa 2 lisäystä ja 2 poistoa
  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)