Skip to content

Affine arg not passed to BFBatchNorm2d #2

@Aonnghus

Description

@Aonnghus

Hello,
I'd like to use your implementation of BFBatchNorm2d.
As I read the code, I observed that affine=True appears in the arguments of the constructor, but is not stored in an instance attribute self.affine nor passed to the super().__init__ method, whereas it is the case for num_features, eps and momentum.
self.affine is however used in the forward method.
If I am not mistaken, it means that passing affine=False to BFBatchNorm2d would not do anything?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions