소스 검색

Update GraphKeys.VARIABLES to GraphKeys.GLOBAL_VARIABLES

Christopher Shallue 9 년 전
부모
커밋
2c83637db7
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  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,
         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: