Internal change

PiperOrigin-RevId: 364625895

Internal change
PiperOrigin-RevId: 364625895
fd2a02af · Le Hou · A. Unique TensorFlower · a08b2932 · fd2a02af
Commit fd2a02af authored Mar 23, 2021 by Le Hou Committed by A. Unique TensorFlower Mar 23, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/utils/testing/mock_task.py official/utils/testing/mock_task.py +1 -1

No files found.
--- a/official/utils/testing/mock_task.py
+++ b/official/utils/testing/mock_task.py
@@ -51,7 +51,7 @@ class MockTask(base_task.Task):
  def build_model(self, *arg, **kwargs):
    inputs = tf.keras.layers.Input(shape=(2,), name="random", dtype=tf.float32)
    outputs = tf.keras.layers.Dense(
-        1, bias_initializer=tf.keras.initializers.Ones())(
+        1, bias_initializer=tf.keras.initializers.Ones(), name="dense_0")(
            inputs)
    network = tf.keras.Model(inputs=inputs, outputs=outputs)
    return MockModel(network)