How do I count the parameters of the model to calculate the complexity of the model? - Githubissues

meijieru / crnn.pytorch

Convolutional recurrent network in pytorch

MIT License

2.4k stars 657 forks source link

How do I count the parameters of the model to calculate the complexity of the model? #215

Open happy20200 opened 4 years ago

Alexgolshtein commented 4 years ago

Try to use Torch-Summary https://github.com/sksq96/pytorch-summary