bug fixed: for loss and accuracy calculation in train.py

[Yeaseen]

Torch version: 1.12.0+cu102
Google Colab CPU runtime.
First error:

so, torch._C._nn.cross_entropy_loss function accepts a tensor.
solve: pass the tensor while calculating loss.

Then, another error occurred:

solve: the loss variable must be float type as it handles float type data

Then, another error occurred:

So, again, pass the tensor while calculating accuracy:

Finally, it's running successfully: