wzzzd / lm_ner

基于Pytorch的命名实体识别框架,支持LSTM+CRF、Bert+CRF、RoBerta+CRF等框架
72 stars 18 forks source link

求解答!!! #17

Open khazic opened 1 year ago

khazic commented 1 year ago

想问一下怎么开多卡呀 我把那个可见的gpu=【1,2,3,0】,但是跑起来的还是只有1 gpu 为什么阿

wzzzd commented 1 year ago

Config.py文件,修改参数self.visible_device = '0,1,2,3'

khazic commented 1 year ago

我就是这么修改的 我看了后台gpu 并没有一起跑

khazic commented 1 year ago

哥 还有个问题想请教一下 就是我output出来的文本带【unk】 这个是vocabe上没有的词 那我怎么才能把unk替换成text原本的词呢 比如我输入的文本是hamburger 但是vocabe里面没有这个词 他就自动识别成了unkmow 然后output里面出来的也是unk 可以换成原始文本hamburger吗