Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "ee5667c1ff7bbc0f4edfd661b033b61fc6dbfbec"
Commit 692da0ae authored by Le Hou's avatar Le Hou Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 364625895
parent 9920d1b9
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/utils/testing/mock_task.py
View file @ 692da0ae
...@@ -51,7 +51,7 @@ class MockTask(base_task.Task): ...@@ -51,7 +51,7 @@ class MockTask(base_task.Task):
def build_model(self, *arg, **kwargs): def build_model(self, *arg, **kwargs):
inputs = tf.keras.layers.Input(shape=(2,), name="random", dtype=tf.float32) inputs = tf.keras.layers.Input(shape=(2,), name="random", dtype=tf.float32)
outputs = tf.keras.layers.Dense( outputs = tf.keras.layers.Dense(
1, bias_initializer=tf.keras.initializers.Ones())( 1, bias_initializer=tf.keras.initializers.Ones(), name="dense_0")(
inputs) inputs)
network = tf.keras.Model(inputs=inputs, outputs=outputs) network = tf.keras.Model(inputs=inputs, outputs=outputs)
return MockModel(network) return MockModel(network)
......
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