(self, x)
| 70 | self.relu2 = torch.nn.ReLU() |
| 71 | |
| 72 | def forward(self, x): |
| 73 | a = self.linear1(x) |
| 74 | a = self.relu1(a) |
| 75 | a = torch.sigmoid(a) |
| 76 | b = self.linear2(a) |
| 77 | b = self.relu2(b) |
| 78 | return b |
| 79 | |
| 80 | |
| 81 | class Model(torch.nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected