search

taishan1994 / BERT-BILSTM-CRF

使用BERT-BILSTM-CRF进行中文命名实体识别。
291 stars 32 forks source link

TypeError #1

Open ccc0917 opened 1 year ago

ccc0917 commented 1 year ago

Traceback (most recent call last): File "D:\python\python-workspace\BERT-BILSTM-CRF\main.py", line 189, in main(data_name) File "D:\python\python-workspace\BERT-BILSTM-CRF\main.py", line 159, in main model = BertNer(args) File "D:\python\python-workspace\BERT-BILSTM-CRF\model.py", line 25, in init self.crf = CRF(args.num_labels, batch_first=True) TypeError: init() got an unexpected keyword argument 'batch_first' 大佬们，这个是什么问题呢？怎么解决啊

taishan1994 commented 1 year ago

Traceback (most recent call last): File "D:\python\python-workspace\BERT-BILSTM-CRF\main.py", line 189, in main(data_name) File "D:\python\python-workspace\BERT-BILSTM-CRF\main.py", line 159, in main model = BertNer(args) File "D:\python\python-workspace\BERT-BILSTM-CRF\model.py", line 25, in init self.crf = CRF(args.num_labels, batch_first=True) TypeError: init() got an unexpected keyword argument 'batch_first' 大佬们，这个是什么问题呢？怎么解决啊

pip install pytorch-crf==0.7.2