(self, x)
| 115 | self.fc3 = nn.Linear(l2, 10) |
| 116 | |
| 117 | def forward(self, x): |
| 118 | x = self.pool(F.relu(self.conv1(x))) |
| 119 | x = self.pool(F.relu(self.conv2(x))) |
| 120 | x = torch.flatten(x, 1) # flatten all dimensions except batch |
| 121 | x = F.relu(self.fc1(x)) |
| 122 | x = F.relu(self.fc2(x)) |
| 123 | x = self.fc3(x) |
| 124 | return x |
| 125 | |
| 126 | ###################################################################### |
| 127 | # Define the search space |
nothing calls this directly
no outgoing calls
no test coverage detected