(self, inputs, targets)
| 54 | self.nll_loss = nn.NLLLoss2d(weight, size_average, ignore_index) |
| 55 | |
| 56 | def forward(self, inputs, targets): |
| 57 | return self.nll_loss((1 - F.softmax(inputs)) ** self.gamma * F.log_softmax(inputs), targets) |
| 58 | |
| 59 | |
| 60 | def _fast_hist(label_pred, label_true, num_classes): |
nothing calls this directly
no outgoing calls
no test coverage detected