diff --git a/examples/transformer/train.py b/examples/transformer/train.py index 41e2786b1..b876d1ae1 100644 --- a/examples/transformer/train.py +++ b/examples/transformer/train.py @@ -177,7 +177,7 @@ def main(_): if step % LOG_EVERY == 0: steps_per_sec = LOG_EVERY / (time.time() - prev_time) prev_time = time.time() - metrics |= {'steps_per_sec': steps_per_sec} + metrics |= {'steps_per_sec': steps_per_sec} # pyrefly: ignore[unsupported-operation] logging.info({k: float(v) for k, v in metrics.items()})