Explorar o código

Merge pull request #718 from cshallue/master

Update GraphKeys.VARIABLES to GraphKeys.GLOBAL_VARIABLES
Chris Shallue %!s(int64=8) %!d(string=hai) anos
pai
achega
a83ef502d9
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      im2txt/im2txt/show_and_tell_model.py

+ 1 - 1
im2txt/im2txt/show_and_tell_model.py

@@ -192,7 +192,7 @@ class ShowAndTellModel(object):
         trainable=self.train_inception,
         is_training=self.is_training())
     self.inception_variables = tf.get_collection(
-        tf.GraphKeys.VARIABLES, scope="InceptionV3")
+        tf.GraphKeys.GLOBAL_VARIABLES, scope="InceptionV3")
 
     # Map inception output into embedding space.
     with tf.variable_scope("image_embedding") as scope: