(self, *args, norm: str = 'none',
norm_kwargs: tp.Dict[str, tp.Any] = {}, **kwargs)
| 127 | to provide a uniform interface across normalization approaches. |
| 128 | """ |
| 129 | def __init__(self, *args, norm: str = 'none', |
| 130 | norm_kwargs: tp.Dict[str, tp.Any] = {}, **kwargs): |
| 131 | super().__init__() |
| 132 | self.conv = apply_parametrization_norm(nn.Conv2d(*args, **kwargs), norm) |
| 133 | self.norm = get_norm_module(self.conv, causal=False, norm=norm, **norm_kwargs) |
| 134 | self.norm_type = norm |
| 135 | |
| 136 | def forward(self, x): |
| 137 | x = self.conv(x) |
nothing calls this directly
no test coverage detected