From eb5d4cb6efd238a0f30ef8b885a5873093307405 Mon Sep 17 00:00:00 2001 From: Stefan Schweter Date: Tue, 26 Sep 2017 22:50:12 +0200 Subject: [PATCH] model_builder: fix log message for diet variables --- tensor2tensor/utils/model_builder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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")