pytorch-summary Add batchnorm parameters: running_mean, running

Add batchnorm parameters: running_mean, running_var, and others

Open mostafaelhoushi opened this issue 5 years ago • 1 comments

Batchnorm has other parameters - in addition to weight and bias - that are not counted in torchsummary, such as running_mean and running_var:

https://pytorch.org/docs/stable/_modules/torch/nn/modules/batchnorm.html

Aug 08 '19 22:08 mostafaelhoushi

I have created a pull request to address this issue: https://github.com/sksq96/pytorch-summary/pull/88

Aug 09 '19 19:08 mostafaelhoushi