Skip to content

GitLab

Commit a83ef502 authored by Chris Shallue's avatar Chris Shallue Committed by GitHub
Browse files

Merge pull request #718 from cshallue/master 

Update GraphKeys.VARIABLES to GraphKeys.GLOBAL_VARIABLES
parents 4bcbe8e1 2c83637d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
im2txt/im2txt/show_and_tell_model.py
View file @ a83ef502
...@@ -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