When I am trying to fit a model I am getting error : Expected object of type torch.cuda.LongTensor but found type torch.cuda.IntTensor


(ritika) #1

When I am trying to fit a model I am getting below error
RuntimeError: Expected object of type torch.cuda.LongTensor but found type torch.cuda.IntTensor for argument #2 ‘target’

Anyone please suggest how to fix this issue.

I am using fastai Version: 1.0.36.post1 and windows 10

Thanks,
Ritika