加载不同模型出错

taishan1994 / pytorch_bert_bilstm_crf_ner

基于pytorch的bert_bilstm_crf中文命名实体识别

507 stars 79 forks source link

Open bjf521zxc opened 4 months ago

bjf521zxc commented 4 months ago

在训练不同模型的时候会出错。同一个虚拟环境下，bert-bilstm-crf可以训练并且也有结果，但把bilstm设置为False就会出现runtimeError。请问一下作者这个会是什么原因呢？

taishan1994 commented 4 months ago

bilstm设置为false的时候有没有把crf设置为true

bjf521zxc commented 4 months ago

设置了，但是依旧报错RuntimeError: Triton Error [CUDA]: invalid device context。网上好像也找不到类似问题