Add total params to weights_summary table #4222
Labels
duplicate
This issue or pull request already exists
feature
Is an improvement or enhancement
help wanted
Open to be worked on
won't fix
This will not be worked on
🚀 Feature
Add total number of parameters when printing the weights_summary table.
Motivation
Since the total number of parameters for each layer is already calculated, it would be really informative if a total sum of number of parameters were also provided. Something like https://github.com/TylerYep/torch-summary or https://github.com/nmhkahn/torchsummaryX provide in their table.
Pitch
When printing weights_summary, there should also appear a summary of the whole model information, something like:
Alternatives
Maybe use one of the given tools (https://github.com/TylerYep/torch-summary or https://github.com/nmhkahn/torchsummaryX) inside?
Additional context
Example from https://github.com/TylerYep/torch-summary model summary:
Example from https://github.com/nmhkahn/torchsummaryX model summary:
The text was updated successfully, but these errors were encountered: