Method
__init__
(self, in_channel, out_channel, kernel_size, stride=1, padding=0, bias=True)
Source from the content-addressed store, hash-verified
| 124 | |
| 125 | class EqualConv2d(nn.Module): |
| 126 | def __init__(self, in_channel, out_channel, kernel_size, stride=1, padding=0, bias=True): |
| 127 | super().__init__() |
| 128 | |
| 129 | self.weight = nn.Parameter(torch.randn(out_channel, in_channel, kernel_size, kernel_size)) |
| 130 | self.scale = 1 / math.sqrt(in_channel * kernel_size ** 2) |
| 131 | |
| 132 | self.stride = stride |
| 133 | self.padding = padding |
| 134 | |
| 135 | if bias: |
| 136 | self.bias = nn.Parameter(torch.zeros(out_channel)) |
| 137 | else: |
| 138 | self.bias = None |
| 139 | |
| 140 | def forward(self, input): |
| 141 | |
Callers
nothing calls this directly
Tested by
no test coverage detected