RuntimeError: expected scalar type Long but found Int #71
Description
D:\anaconda3\envs\pytorch114\lib\site-packages\torch\optim\lr_scheduler.py:134: UserWarning: Detected call of lr_scheduler.step() before `optimizer.st
File "train.py", line 202, in
train_model()
File "train.py", line 137, in train_model
loss = criterion(outputs, labels)
File "D:\anaconda3\envs\pytorch114\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl
result = self.forward(*input, **kwargs)
File "D:\anaconda3\envs\pytorch114\lib\site-packages\torch\nn\modules\loss.py", line 1048, in forward
ignore_index=self.ignore_index, reduction=self.reduction)
File "D:\anaconda3\envs\pytorch114\lib\site-packages\torch\nn\functional.py", line 2690, in cross_entropy
return nll_loss(log_softmax(input, 1), target, weight, None, ignore_index, None, reduction)
File "D:\anaconda3\envs\pytorch114\lib\site-packages\torch\nn\functional.py", line 2385, in nll_loss
ret = torch._C._nn.nll_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index)
RuntimeError: expected scalar type Long but found Int