Fix the test failure caused by keras initializer change.

PiperOrigin-RevId: 454463130

Fix the test failure caused by keras initializer change.
PiperOrigin-RevId: 454463130
2c254c63 · Scott Zhu · A. Unique TensorFlower · c2a8f717 · 2c254c63
Commit 2c254c63 authored Jun 12, 2022 by Scott Zhu Committed by A. Unique TensorFlower Jun 12, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

official/vision/serving/detection_test.py official/vision/serving/detection_test.py +4 -0

No files found.
--- a/official/vision/serving/detection_test.py
+++ b/official/vision/serving/detection_test.py
@@ -29,6 +29,10 @@ from official.vision.serving import detection

 class DetectionExportTest(tf.test.TestCase, parameterized.TestCase):

+  def setUp(self):
+    super().setUp()
+    tf.keras.utils.set_random_seed(1)
+
  def _get_detection_module(self, experiment_name, input_type):
    params = exp_factory.get_exp_config(experiment_name)
    params.task.model.backbone.resnet.model_id = 18