Skip to content

GitLab

"examples/git@developer.sourcefind.cn:change/sglang.git" did not exist on "9f81d741a28667f05037d14c83491a740fb2251a"
Commit fd2a02af authored by Le Hou's avatar Le Hou Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 364625895
parent a08b2932
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/utils/testing/mock_task.py
View file @ fd2a02af
...@@ -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