Skip to content

Commit

Permalink
minor model summary fix
Browse files Browse the repository at this point in the history
  • Loading branch information
kartik4949 committed Jan 20, 2021
1 parent 63010a8 commit eb6282b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pytorch_lightning/core/memory.py
Expand Up @@ -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

Expand Down

0 comments on commit eb6282b

Please sign in to comment.