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