(self, x)
| 56 | self.softmax = torch.nn.Softmax(dim=1) |
| 57 | |
| 58 | def forward(self, x): |
| 59 | x = self.linear1(x) |
| 60 | x = self.activation(x) |
| 61 | x = self.linear2(x) |
| 62 | x = self.softmax(x) |
| 63 | return x |
| 64 | |
| 65 | tinymodel = TinyModel() |
| 66 |
nothing calls this directly
no outgoing calls
no test coverage detected