dbiir / UER-py

Open Source Pre-training Model Framework in PyTorch & Pre-trained Model Zoo
https://github.com/dbiir/UER-py/wiki
Apache License 2.0
3.01k stars 525 forks source link

修复ELMO预训练中bilm_target.py的seg参数缺失问题 #404

Open liu673 opened 2 months ago

liu673 commented 2 months ago

在预训练ELMO模型时,我发现bilm_target.py中的前向传播和后向传播函数调用缺少了seg参数传递,这导致了预训练过程中的运行时错误:TypeError: LmTarget.lm() missing 1 required positional argument: 'seg'

此PR修复了这个问题,通过在相关函数调用中加入seg参数。这样可以确保在进行前向和后向传播时正确地传递所有必需的参数,从而避免上述类型错误。

具体更改:

测试情况:

请审查这些更改,感谢您的时间和努力!