瀏覽代碼

Update GraphKeys.VARIABLES to GraphKeys.GLOBAL_VARIABLES.

Christopher Shallue 9 年之前
父節點
當前提交
946c75e5b5
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      im2txt/im2txt/show_and_tell_model.py

+ 1 - 1
im2txt/im2txt/show_and_tell_model.py

@@ -343,7 +343,7 @@ class ShowAndTellModel(object):
         initial_value=0,
         initial_value=0,
         name="global_step",
         name="global_step",
         trainable=False,
         trainable=False,
-        collections=[tf.GraphKeys.GLOBAL_STEP, tf.GraphKeys.VARIABLES])
+        collections=[tf.GraphKeys.GLOBAL_STEP, tf.GraphKeys.GLOBAL_VARIABLES])
 
 
     self.global_step = global_step
     self.global_step = global_step