run train.py error about torch-light HOT 6 CLOSED

make sure the labels` data type is torch.cuda.LongTensor
i saw your log that something wrong in the labels' datatype (torch.cuda.IntTensor)

File "D:/github/torch_light/Image-Cap/train.py", line 235, in
target, _ = actor(hidden, labels)

from torch-light.

@ne7ermore
hi， I don't modify Image-Cap/train.py， i saw return labels is int list.
thx!

from torch-light.

what did you get from the data_loader. Data_loader
try to get the labels` datatype form this line below i show

https://github.com/ne7ermore/torch_light/blob/master/Image-Cap/data_loader.py#L41

or you can try this simple test code: https://github.com/ne7ermore/torch_light/blob/master/Image-Cap/data_loader.py#L58

img, labels = next(training_data) from https://github.com/ne7ermore/torch_light/blob/master/Image-Cap/data_loader.py#L66
print(labels)

from torch-light.

@ne7ermore
Hi, when I run Image-Cap/data_loader.py, labels is [torch.cuda.IntTensor of size 2x16 (GPU 0)], Image-Cap/train.py#115 labels is [torch.cuda.IntTensor of size 32x17 (GPU 0)]
thx!

from torch-light.

I guess sth change between pytorch`s versions
my version is 0.3.1
and i have added long() function to change the datatype of labels

https://github.com/ne7ermore/torch_light/blob/master/Image-Cap/data_loader.py#L46

please try again :)

if found error again, show the pytorch's version which you used

from torch-light.

@ne7ermore
Thank you so much, it works out! train.py has been running!

from torch-light.