Add KL div as regularization to CrossEntropyLoss

[netaglazer (neta)]

hi
I use CrossEntropyLoss, and I want to add KL div between labels and prediction as regularization to the loss

how can I do it?
thanks

[alpha_omega_alpha]

Check out the VAE loss_function in Pytorch’s examples here.