Skip to content

GitLab

"scripts/convert_original_t2i_adapter.py" did not exist on "39299546134b545c1b78b0937147f85aed8fbe4d"
Commit 9e5ec223 authored by Scott Zhu's avatar Scott Zhu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 371197345
parent fa9c975f
Show 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):
def validation_step(self, inputs, model: tf.keras.Model, metrics=None):
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
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