Getting loss 0.000 and keyerror while inferencing

[ecdrid (Aditya)]

I hope this will help you understand the error for sure CrossEntropyLoss…