(self, x)
| 91 | self.model = nn.Sequential(*layers) # No need to use Relu for the last layer |
| 92 | |
| 93 | def forward(self, x): |
| 94 | x = self.model(x) |
| 95 | return F.log_softmax(x, dim=1) |
| 96 | |
| 97 | def training_step(self, batch, batch_idx): |
| 98 | x, y = batch |
nothing calls this directly
no outgoing calls
no test coverage detected