bert_chinese_pytorch icon indicating copy to clipboard operation
bert_chinese_pytorch copied to clipboard

报错__init__() missing 1 required positional argument: 'num_labels'

Open Lingzhi-WANG opened this issue 5 years ago • 1 comments

Traceback (most recent call last): File "bert.py", line 651, in main() File "bert.py", line 541, in main cache_dir=PYTORCH_PRETRAINED_BERT_CACHE / 'distributed_{}'.format(args.local_rank)) File "/home/lindseywang/.local/lib/python3.6/site-packages/pytorch_pretrained_bert/modeling.py", line 581, in from_pretrained model = cls(config, *inputs, **kwargs) TypeError: init() missing 1 required positional argument: 'num_labels'

请问这是因为什么呢?

Lingzhi-WANG avatar Apr 28 '19 01:04 Lingzhi-WANG

@LingzhiWang2017 在model中结尾添加num_labels=len(label_list)即可

Zhangpeixiang avatar Jun 27 '19 10:06 Zhangpeixiang