Skip to content

Add batchnorm parameters: running_mean, running_var, and others #87

@mostafaelhoushi

Description

@mostafaelhoushi

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions