Thank you so much.I I did the changes, as advised by you. I just came across with one error after that which is RuntimeError: 1D target tensor expected, multi-target not supported
SO for that I updated the line of code as below:
loss = criterion(output, labels.squeeze())
The code is on Training since an half an hour,
can you please provide me solution for decreasing  the training time  and “1D target tensor expected, multi-target not supported” error.
I greatly appreciate your help