Improve error message: .numpy() not supported on cuda
I was struggling to feed my predictions into an sklearn function <SCODE>M_ = M.data.numpy()
<ECODE> I got error <SCODE>RuntimeError: numpy conversion for LongTensor is not supported
<ECODE> <SCODE>M_ = M.data.cpu().numpy()
<ECODE> I suggest to improve this error message. Maybe something like <SCODE>RuntimeError: numpy conversion for LongTensor is not supported on CUDA
<ECODE> That way, novices like me will know what to do
robromijnders
thanks, it’s a good suggestion. i’ll work on this.
smth