Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/ktransformers.git" did not exist on "877aec858e3885c956955b9aade3d38bd0ee6214"
Commit 26d8e705 authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 405455038
parent 6cd536dc
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
official/vision/beta/serving/export_module_factory_test.py
View file @ 26d8e705
...@@ -107,8 +107,10 @@ class ImageClassificationExportTest(tf.test.TestCase, parameterized.TestCase): ...@@ -107,8 +107,10 @@ class ImageClassificationExportTest(tf.test.TestCase, parameterized.TestCase):
# The imported model should contain any trackable attrs that the original # The imported model should contain any trackable attrs that the original
# model had. # model had.
self.assertTrue(hasattr(imported.model, 'test_trackable')) self.assertTrue(hasattr(imported.model, 'test_trackable'))
self.assertAllClose(out['logits'].numpy(), expected_logits.numpy()) self.assertAllClose(
self.assertAllClose(out['probs'].numpy(), expected_prob.numpy()) out['logits'].numpy(), expected_logits.numpy(), rtol=1e-04, atol=1e-04)
self.assertAllClose(
out['probs'].numpy(), expected_prob.numpy(), rtol=1e-04, atol=1e-04)
if __name__ == '__main__': if __name__ == '__main__':
......
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