The loses on final linear layer of CNN

[tojiboyevf]

Hi!

Thanks for your work! I would like to clarify which loses you use at the final linear layer of CNN. As I understand, you use both L2 and KL divergence