ShannonAI / mrc-for-flat-nested-ner

Code for ACL 2020 paper `A Unified MRC Framework for Named Entity Recognition`

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CUDA11运行改代码

Josson opened this issue · comments

commented

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

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

commented

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

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

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

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

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

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

好的,感谢您的回复!