Skip to content

GitLab

Commit 32dadfc2 authored by David Chen's avatar David Chen Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 269914707
parent c4bacbf8
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/transformer/v2/transformer_main.py
View file @ 32dadfc2
...@@ -180,10 +180,10 @@ class TransformerTask(object): ...@@ -180,10 +180,10 @@ class TransformerTask(object):
if not params["static_batch"]: if not params["static_batch"]:
raise ValueError("TPU requires static batch for input data.") raise ValueError("TPU requires static batch for input data.")
else: else:
logging.info("Running transformer with num_gpus =", num_gpus) logging.info("Running transformer with num_gpus = %d", num_gpus)
if self.distribution_strategy: if self.distribution_strategy:
logging.info("For training, using distribution strategy: ", logging.info("For training, using distribution strategy: %s",
self.distribution_strategy) self.distribution_strategy)
else: else:
logging.info("Not using any distribution strategy.") logging.info("Not using any distribution strategy.")
......
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