Runtime Error: Expected object of scalar type Long but got scalar type Byte for argument #2 'target'

[immkapoor]

Loading Dataset
Start training
Traceback (most recent call last):
File "train_cnn_single_frame.py", line 59, in
validate=True, resume=resume, use_cuda=cuda)
File "D:\Action Recognition\KTH Program\KTH-Action-Recognition-master\KTH-Action-Recognition-master\main\train_helper.py", line 109, in train
loss = criterion(outputs, labels)
File "C:\Users\Meghna\Anaconda\envs\tensorflow_env\lib\site-packages\torch\nn\modules\module.py", line 489, in call
result = self.forward(*input, **kwargs)
File "C:\Users\Meghna\Anaconda\envs\tensorflow_env\lib\site-packages\torch\nn\modules\loss.py", line 904, in forward
ignore_index=self.ignore_index, reduction=self.reduction)
File "C:\Users\Meghna\Anaconda\envs\tensorflow_env\lib\site-packages\torch\nn\functional.py", line 1970, in cross_entropy
return nll_loss(log_softmax(input, 1), target, weight, None, ignore_index, None, reduction)
File "C:\Users\Meghna\Anaconda\envs\tensorflow_env\lib\site-packages\torch\nn\functional.py", line 1790, in nll_loss
ret = torch._C._nn.nll_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index)
RuntimeError: Expected object of scalar type Long but got scalar type Byte for argument #2 'target'

[ashuezy]

Add this line

labels = labels.long()

in the train method before calling criterion(....

[immkapoor]

@ashuezy Hi, I tried adding this line yet I am getting same error.

[jaycelol10]

@ashuezy @vkhoi @immkapoor can you please tell me how u solved it guys ?