(self, x)
| 23 | self.sigmoid = nn.Sigmoid() |
| 24 | |
| 25 | def forward(self, x): |
| 26 | x = x.permute(0, 3, 1, 2) |
| 27 | x = self.leakyrelu1(self.bn1(self.conv1(x))) |
| 28 | x = self.leakyrelu2(self.bn2(self.conv2(x))) |
| 29 | x = self.leakyrelu3(self.bn3(self.conv3(x))) |
| 30 | x = torch.flatten(x, 1) |
| 31 | x = self.sigmoid(self.fc(x)) |
| 32 | return x |
| 33 | |
| 34 | class model(nn.Module): |
| 35 | def __init__(self, num_basis): |
nothing calls this directly
no outgoing calls
no test coverage detected