I am training a Resent-18 model on CIFAR- 10 dataset. But my model is failing to learn over label1 and 5 after certain epochs.
One can see the performance of the model on: https://wandb.ai/praveenjoshi007/sfl/reports/Shared-panel-21-10-27-22-10-26--VmlldzoxMTYyOTI3?accessToken=6einm83ach3zd2zsl2cowel8o2n7t9k9gg64bqs5coaoryql0a0qtsrzoegqucbk
Is there any function that can help make my model learn on the labels showing a high loss or having low classification accuracy?