706fdc0984a1016bc7f9616cc988dc4b53ea3793,torch_geometric/nn/norm/batch_norm.py,BatchNorm,__repr__,#BatchNorm#,39

Before Change

        return super(BatchNorm, self).forward(x)

    def __repr__(self):
        return ("{}({}, eps={}, momentum={}, affine={}, "
                "track_running_stats={})").format(self.__class__.__name__,
                                                  self.num_features, self.eps,
                                                  self.momentum, self.affine,
                                                  self.track_running_stats)

After Change

        return self.module(x)

    def __repr__(self):
        return f"{self.__class__.__name__}({self.module.num_features})"

Instances