Expected object of backend CPU but got backend CUDA for argument #3 'index'

[wlsrick]

I've got this problem in line 51
alpha = self.alpha.gather(0,labels.view(-1))

How can I solve it?
thanks a lot.