I was trying to build a multi class classification model on CIFAR-10 dataset and I was trying to build a loss function that was mentioned in this paper: Improved Loss Function for Image Classification
However, it seems that there is something wrong with my implementation.
Can someone help me, please?
This is the colab link: Google Colab
Thanks