pytorch-classification-uncertainty
pytorch-classification-uncertainty copied to clipboard
why + torch.lgamma(ones).sum(dim=1, keepdim=True) in kl_divergence?
Torch.lgamma(1)=0 and no such term in the original paper