def compute_loss()

in trainer/loss.py [0:0]

• 2 lines of code
• 1 McCabe index (conditional complexity)

    def compute_loss(self, logits, mask):
        return - torch.log( (F.softmax(logits, dim=1) * mask).sum(1) )