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
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions