search

wzzzd / text_classifier_pytorch

基于Pytorch的文本分类框架，支持TextCNN、Bert、Electra等。
57 stars 9 forks source link

测试时报错 #6

Closed huobenhuo closed 1 year ago

huobenhuo commented 1 year ago

mode = 'test'时报错： File "D:\text_classifier_pytorch\module\Predictor.py", line 124, in predict output = self.model(**batch) TypeError: init() got an unexpected keyword argument 'label'

怎么解决呀

wzzzd commented 1 year ago

你用的是哪个模型？

huobenhuo commented 1 year ago

你用的是哪个模型？

TextRNN

huobenhuo commented 1 year ago

你用的是哪个模型？

补充一下，FastText/TextCNN/TextRCNN/Transformer也都存在这个问题

huobenhuo commented 1 year ago

您好这个问题解决了，是我这里的路径问题，给您添麻烦了

wzzzd commented 1 year ago

解决了就好~