(self)
| 253 | class TinyModel(torch.nn.Module): |
| 254 | |
| 255 | def __init__(self): |
| 256 | super(TinyModel, self).__init__() |
| 257 | |
| 258 | self.layer1 = torch.nn.Linear(DIM_IN, HIDDEN_SIZE) |
| 259 | self.relu = torch.nn.ReLU() |
| 260 | self.layer2 = torch.nn.Linear(HIDDEN_SIZE, DIM_OUT) |
| 261 | |
| 262 | def forward(self, x): |
| 263 | x = self.layer1(x) |
nothing calls this directly
no outgoing calls
no test coverage detected