How many classes are you using?
The loss value seem to be quite high for a reasonable accuracy, but might of course still be valid.
If you don’t reduce the loss, you would have to pass the gradient into backward with the same shape as your output.
The usual approach of printing some statistics is to use a condition inside your training loop as seen e.g. here.