Dhyanil Mehta
Results
1
comments of
Dhyanil Mehta
You can add .long() to batch_labels.view(-1,1) in the line of code where the error is coming. Then the error will go away.