Skip to content

GitLab

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

Internal change 

PiperOrigin-RevId: 371197345
parent fa9c975f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/utils/testing/mock_task.py
View file @ 9e5ec223
...@@ -62,7 +62,7 @@ class MockTask(base_task.Task): ...@@ -62,7 +62,7 @@ class MockTask(base_task.Task):
def validation_step(self, inputs, model: tf.keras.Model, metrics=None): def validation_step(self, inputs, model: tf.keras.Model, metrics=None):
logs = super().validation_step(inputs, model, metrics) logs = super().validation_step(inputs, model, metrics)
logs["counter"] = tf.ones((1,), dtype=tf.float32) logs["counter"] = tf.constant(1, dtype=tf.float32)
return logs return logs
def build_inputs(self, params): def build_inputs(self, params):
......
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