diff --git a/tensor2tensor/utils/model_builder.py b/tensor2tensor/utils/model_builder.py index 4a4717bd4..6e0b32b13 100644 --- a/tensor2tensor/utils/model_builder.py +++ b/tensor2tensor/utils/model_builder.py @@ -288,7 +288,7 @@ def nth_model(n): diet_vars = [ v for v in tf.global_variables() if v.dtype == dtypes.float16_ref ] - _log_variable_sizes(diet_vars, "Diet Varaibles") + _log_variable_sizes(diet_vars, "Diet Variables") # Optimize total_loss = tf.identity(total_loss, name="total_loss")