TylerYep / torchinfo

View model summaries in PyTorch!
MIT License
2.56k stars 119 forks source link

Idea: add the column "Params %" in addition to "Params #" #159

Closed Yura52 closed 1 year ago

Yura52 commented 2 years ago

I use torchinfo to get a rough understanding of model sizes and how the "weight" is distributed across the model. Unfortunately, the latter information is not available directly, which is not convenient, especially with larger models. So I suggest adding one more column which contains the ratio n_params / total_n_params. It would immediately provide new important insights almost for free.

TylerYep commented 2 years ago

This is a good idea! PRs adding this functionality are welcome (the functionality should be fairly straightforward)