Skip to content

GitLab

".github/git@developer.sourcefind.cn:change/sglang.git" did not exist on "522e18eaebc2f14135249bf95f24ce3505683082"
Commit a671a5a6 authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 405455038
parent 2492f768
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 @ a671a5a6
...@@ -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