(self,x)
| 42 | torch.nn.init.zeros_(self.bias) |
| 43 | |
| 44 | def forward(self,x): |
| 45 | out = self.conv(x * self.scale) + self.bias.view(1,self.bias.shape[0],1,1) |
| 46 | # print(out.shape) |
| 47 | return out |
| 48 | |
| 49 | class WSLinear(torch.nn.Module): |
| 50 | def __init__(self,in_features,out_features,gain = 2): |
nothing calls this directly
no outgoing calls
no test coverage detected