Skip to content

GitLab

"examples/vscode:/vscode.git/clone" did not exist on "bdbcaa9852565d3854100f6504266ffcc7405b05"
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
Show 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