(
self,
num_groups: int,
num_channels: int,
eps: float = 1e-05,
affine: bool = True,
device=None,
dtype=None,
)
| 116 | """ |
| 117 | |
| 118 | def __init__( |
| 119 | self, |
| 120 | num_groups: int, |
| 121 | num_channels: int, |
| 122 | eps: float = 1e-05, |
| 123 | affine: bool = True, |
| 124 | device=None, |
| 125 | dtype=None, |
| 126 | ) -> None: |
| 127 | super().__init__() |
| 128 | assert num_groups > 0, "The num_groups must larger than zero" |
| 129 | assert num_channels > 0, "The num_channels must larger than zero" |
| 130 | self.num_groups = num_groups |
| 131 | self.num_channels = num_channels |
| 132 | self.eps = eps |
| 133 | self.affine = affine |
| 134 | factory_kwargs = {} |
| 135 | if device: |
| 136 | factory_kwargs["device"] = device |
| 137 | if dtype: |
| 138 | factory_kwargs["dtype"] = dtype |
| 139 | if self.affine: |
| 140 | self.weight = flow.nn.Parameter(flow.Tensor(num_channels, **factory_kwargs)) |
| 141 | self.bias = flow.nn.Parameter(flow.Tensor(num_channels, **factory_kwargs)) |
| 142 | else: |
| 143 | self.register_parameter("weight", None) |
| 144 | self.register_parameter("bias", None) |
| 145 | self.reset_parameters() |
| 146 | |
| 147 | def reset_parameters(self) -> None: |
| 148 | if self.affine: |
nothing calls this directly
no test coverage detected