(self, x)
| 93 | return nn.Sequential(*layers) |
| 94 | |
| 95 | def forward(self, x): |
| 96 | x = self.conv1(x) |
| 97 | x = self.bn1(x) |
| 98 | x = self.relu(x) |
| 99 | |
| 100 | x = self.layer1(x) |
| 101 | x = self.layer2(x) |
| 102 | x = self.layer3(x) |
| 103 | x = self.layer4(x) |
| 104 | x = torch.mean(x, dim=2, keepdim=True) |
| 105 | x = x.view((x.size()[0], x.size()[1], -1)) |
| 106 | x = x.transpose(1, 2) |
| 107 | |
| 108 | return x |
nothing calls this directly
no outgoing calls
no test coverage detected