Function gentropy

model.py:478–482

(softmax_out)

def gentropy(softmax_out):
    epsilon = 1e-5
    msoftmax = softmax_out.mean(dim=0)
    gentropy = -msoftmax * torch.log(msoftmax + epsilon)
    return torch.sum(gentropy)


def ent_loss(out):