search

ShannonAI / mrc-for-flat-nested-ner

Code for ACL 2020 paper `A Unified MRC Framework for Named Entity Recognition`
643 stars 117 forks source link

CUDA11运行改代码 #99

Open Josson opened 2 years ago

Josson commented 2 years ago

代码用的是pytorch1.7，但是CUDA11对应的pytorch是1.8的，而服务器只能装CUDA11，请问这种情况怎么运行啊？

Senwang98 commented 2 years ago

@Josson 我也是cuda11，似乎没有什么问题训练。 顺便问一下您之后如果跑通之后精度能复现对齐吗，多个数据集我都训过，精度总是差一点点，大概一个点

Josson commented 2 years ago

@Senwang98 我跑了几个数据集但是差的不止一个点，请问方便交流一下吗

Senwang98 commented 2 years ago

@Josson 挺奇怪，我其实是直接用的提供的脚本训练的。 wx：17714924649

zhanghaok commented 1 year ago

@Senwang98 我跑了几个数据集但是差的不止一个点，请问方便交流一下吗

您好，请问你的pytorch版本是多少呢

Senwang98 commented 1 year ago

@zhanghaok 我个人习惯是只使用pt1.7（17我从来没遇到什么bug），没记错的话就是1.7版本，你试试看

zhanghaok commented 1 year ago

@zhanghaok 我个人习惯是只使用pt1.7（17我从来没遇到什么bug），没记错的话就是1.7版本，你试试看

好的，感谢您的回复！