UER-py icon indicating copy to clipboard operation
UER-py copied to clipboard

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

Open liu673 opened this issue 1 year ago • 0 comments

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

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

具体更改:

  • 在调用self.lm()方法时添加了缺失的seg参数。

测试情况:

  • 已经在我的开发环境中进行了基本的功能测试,预训练过程现在能够正常执行而不再抛出错误。

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

liu673 avatar Sep 12 '24 12:09 liu673