From eb6282bcddd30be335dfe0f52b6648eeb3be35ab Mon Sep 17 00:00:00 2001 From: Kartik Sharma Date: Wed, 20 Jan 2021 19:15:46 +0530 Subject: [PATCH] minor model summary fix --- pytorch_lightning/core/memory.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/core/memory.py b/pytorch_lightning/core/memory.py index de551cabd30df8..d8421086e3008e 100644 --- a/pytorch_lightning/core/memory.py +++ b/pytorch_lightning/core/memory.py @@ -330,7 +330,7 @@ def _format_summary_table(total_parameters: int, trainable_parameters: int, mode summary += "\n" + s.format(get_human_readable_count(total_parameters), 10) summary += "Total params" summary += "\n" + s.format(get_formatted_model_size(model_size), 10) - summary += "Total Estimated Params Size (MB)" + summary += "Total estimated model params size (MB)" return summary