Pārlūkot izejas kodu

Replace deprecated functions

Christopher Shallue 8 gadi atpakaļ
vecāks
revīzija
f653bd2340
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. 1 1
      im2txt/im2txt/ops/image_embedding_test.py

+ 1 - 1
im2txt/im2txt/ops/image_embedding_test.py

@@ -41,7 +41,7 @@ class InceptionV3Test(tf.test.TestCase):
   def _countInceptionParameters(self):
     """Counts the number of parameters in the inception model at top scope."""
     counter = {}
-    for v in tf.all_variables():
+    for v in tf.global_variables():
       name_tokens = v.op.name.split("/")
       if name_tokens[0] == "InceptionV3":
         name = "InceptionV3/" + name_tokens[1]