Skip to content

GitLab

Commit 2c83637d authored by Christopher Shallue's avatar Christopher Shallue
Browse files

Update GraphKeys.VARIABLES to GraphKeys.GLOBAL_VARIABLES

parent 4bcbe8e1
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
im2txt/im2txt/show_and_tell_model.py
View file @ 2c83637d
...@@ -192,7 +192,7 @@ class ShowAndTellModel(object): ...@@ -192,7 +192,7 @@ class ShowAndTellModel(object):
trainable=self.train_inception, trainable=self.train_inception,
is_training=self.is_training()) is_training=self.is_training())
self.inception_variables = tf.get_collection( self.inception_variables = tf.get_collection(
tf.GraphKeys.VARIABLES, scope="InceptionV3") tf.GraphKeys.GLOBAL_VARIABLES, scope="InceptionV3")
# Map inception output into embedding space. # Map inception output into embedding space.
with tf.variable_scope("image_embedding") as scope: with tf.variable_scope("image_embedding") as scope:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment