Andrew Zhuoer Feng

icon indicating a website link fengzhuoer.com icon indicating an email [email protected]

icon company Tsinghua University icon location Beijing, China icon location Third year PhD in Tsinghua University (2022-2027) @thu-coai . Interested in NLP and knowledge alignment.

Results 1 comments of Andrew Zhuoer Feng

RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'mat1' in call to _th_addmm

The reason might be you are calculating Tensors from different devices. I got this problem when CrossEntropyLoss cpu and cuda Tensors :) use `tensor1.cuda()` to convert a cpu tensor to...