RuntimeError: expected device cuda:2 and dtype Float but got device cuda:2 and dtype Long

z814081807 / DeepNER

天池中药说明书实体识别挑战冠军方案；中文命名实体识别；NER; BERT-CRF & BERT-SPAN & BERT-MRC；Pytorch

922 stars 229 forks source link

Closed penghui-pwld-two closed 3 years ago

penghui-pwld-two commented 3 years ago

您好，很感谢您分享的程序，我（小白）在运行的时候出现如下错误，请问如何解决： RuntimeError: expected device cuda:2 and dtype Float but got device cuda:2 and dtype Long

非常感谢您！

z814081807 commented 3 years ago

您好，很感谢您分享的程序，我（小白）在运行的时候出现如下错误，请问如何解决： RuntimeError: expected device cuda:2 and dtype Float but got device cuda:2 and dtype Long

非常感谢您！

pytorch 版本问题应该