Skip to content

GitLab

Commit e202822a authored by Scott Zhu's avatar Scott Zhu Committed by A. Unique TensorFlower
Browse files

Fix the test failure caused by keras initializer change. 

PiperOrigin-RevId: 454463130
parent cec028b5
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
official/vision/serving/detection_test.py
View file @ e202822a
...@@ -29,6 +29,10 @@ from official.vision.serving import detection ...@@ -29,6 +29,10 @@ from official.vision.serving import detection
class DetectionExportTest(tf.test.TestCase, parameterized.TestCase): 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): def _get_detection_module(self, experiment_name, input_type):
params = exp_factory.get_exp_config(experiment_name) params = exp_factory.get_exp_config(experiment_name)
params.task.model.backbone.resnet.model_id = 18 params.task.model.backbone.resnet.model_id = 18
......
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