Runtimeerror: expected object of type Variable[torch.LongTensor] but found type Variable[torch.ByteTensor] for argument #1 'target'

1

QQ%E6%88%AA%E5%9B%BE20180727222805
QQ%E6%88%AA%E5%9B%BE201807272228051339×231 11.2 KB

pytorch 0.4 is ok! but at pytorch 0.3, i get this error!
hello guys, Does anyone know why? very thanks for reply!!

2

target has to be long type, maybe target = target.long() will help
https://pytorch.org/docs/stable/nn.html?highlight=crossentropy#torch.nn.CrossEntropyLoss

3

thanks for your reply. I wirte my code in early version, but new version of pytorch cannot use. i had try your advice , but it is not work . so I change to the old version of pythorch.