Skip to content

GitLab

Commit a46f826f authored by Christopher Shallue's avatar Christopher Shallue
Browse files

Replace deprecated functions

parent 705acc35
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
im2txt/im2txt/ops/image_embedding_test.py
View file @ a46f826f
...@@ -85,7 +85,7 @@ class InceptionV3Test(tf.test.TestCase): ...@@ -85,7 +85,7 @@ class InceptionV3Test(tf.test.TestCase):
self.assertEqual([self._batch_size, 2048], embeddings.get_shape().as_list()) self.assertEqual([self._batch_size, 2048], embeddings.get_shape().as_list())
self._verifyParameterCounts() self._verifyParameterCounts()
self._assertCollectionSize(376, tf.GraphKeys.VARIABLES) self._assertCollectionSize(376, tf.GraphKeys.GLOBAL_VARIABLES)
self._assertCollectionSize(188, tf.GraphKeys.TRAINABLE_VARIABLES) self._assertCollectionSize(188, tf.GraphKeys.TRAINABLE_VARIABLES)
self._assertCollectionSize(188, tf.GraphKeys.UPDATE_OPS) self._assertCollectionSize(188, tf.GraphKeys.UPDATE_OPS)
self._assertCollectionSize(94, tf.GraphKeys.REGULARIZATION_LOSSES) self._assertCollectionSize(94, tf.GraphKeys.REGULARIZATION_LOSSES)
...@@ -98,7 +98,7 @@ class InceptionV3Test(tf.test.TestCase): ...@@ -98,7 +98,7 @@ class InceptionV3Test(tf.test.TestCase):
self.assertEqual([self._batch_size, 2048], embeddings.get_shape().as_list()) self.assertEqual([self._batch_size, 2048], embeddings.get_shape().as_list())
self._verifyParameterCounts() self._verifyParameterCounts()
self._assertCollectionSize(376, tf.GraphKeys.VARIABLES) self._assertCollectionSize(376, tf.GraphKeys.GLOBAL_VARIABLES)
self._assertCollectionSize(188, tf.GraphKeys.TRAINABLE_VARIABLES) self._assertCollectionSize(188, tf.GraphKeys.TRAINABLE_VARIABLES)
self._assertCollectionSize(0, tf.GraphKeys.UPDATE_OPS) self._assertCollectionSize(0, tf.GraphKeys.UPDATE_OPS)
self._assertCollectionSize(94, tf.GraphKeys.REGULARIZATION_LOSSES) self._assertCollectionSize(94, tf.GraphKeys.REGULARIZATION_LOSSES)
...@@ -111,7 +111,7 @@ class InceptionV3Test(tf.test.TestCase): ...@@ -111,7 +111,7 @@ class InceptionV3Test(tf.test.TestCase):
self.assertEqual([self._batch_size, 2048], embeddings.get_shape().as_list()) self.assertEqual([self._batch_size, 2048], embeddings.get_shape().as_list())
self._verifyParameterCounts() self._verifyParameterCounts()
self._assertCollectionSize(376, tf.GraphKeys.VARIABLES) self._assertCollectionSize(376, tf.GraphKeys.GLOBAL_VARIABLES)
self._assertCollectionSize(0, tf.GraphKeys.TRAINABLE_VARIABLES) self._assertCollectionSize(0, tf.GraphKeys.TRAINABLE_VARIABLES)
self._assertCollectionSize(0, tf.GraphKeys.UPDATE_OPS) self._assertCollectionSize(0, tf.GraphKeys.UPDATE_OPS)
self._assertCollectionSize(0, tf.GraphKeys.REGULARIZATION_LOSSES) self._assertCollectionSize(0, tf.GraphKeys.REGULARIZATION_LOSSES)
...@@ -124,7 +124,7 @@ class InceptionV3Test(tf.test.TestCase): ...@@ -124,7 +124,7 @@ class InceptionV3Test(tf.test.TestCase):
self.assertEqual([self._batch_size, 2048], embeddings.get_shape().as_list()) self.assertEqual([self._batch_size, 2048], embeddings.get_shape().as_list())
self._verifyParameterCounts() self._verifyParameterCounts()
self._assertCollectionSize(376, tf.GraphKeys.VARIABLES) self._assertCollectionSize(376, tf.GraphKeys.GLOBAL_VARIABLES)
self._assertCollectionSize(0, tf.GraphKeys.TRAINABLE_VARIABLES) self._assertCollectionSize(0, tf.GraphKeys.TRAINABLE_VARIABLES)
self._assertCollectionSize(0, tf.GraphKeys.UPDATE_OPS) self._assertCollectionSize(0, tf.GraphKeys.UPDATE_OPS)
self._assertCollectionSize(0, tf.GraphKeys.REGULARIZATION_LOSSES) self._assertCollectionSize(0, tf.GraphKeys.REGULARIZATION_LOSSES)
......
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