(self, inputs, targets)
| 44 | self.nll_loss = nn.NLLLoss2d(weight, size_average, ignore_index) |
| 45 | |
| 46 | def forward(self, inputs, targets): |
| 47 | return self.nll_loss(F.log_softmax(inputs), targets) |
| 48 | |
| 49 | |
| 50 | class FocalLoss2d(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected