Open liu673 opened 2 months ago
在预训练ELMO模型时,我发现bilm_target.py中的前向传播和后向传播函数调用缺少了seg参数传递,这导致了预训练过程中的运行时错误:TypeError: LmTarget.lm() missing 1 required positional argument: 'seg'
bilm_target.py
seg
此PR修复了这个问题,通过在相关函数调用中加入seg参数。这样可以确保在进行前向和后向传播时正确地传递所有必需的参数,从而避免上述类型错误。
具体更改:
self.lm()
测试情况:
请审查这些更改,感谢您的时间和努力!
在预训练ELMO模型时,我发现
bilm_target.py
中的前向传播和后向传播函数调用缺少了seg
参数传递,这导致了预训练过程中的运行时错误:TypeError: LmTarget.lm() missing 1 required positional argument: 'seg'此PR修复了这个问题,通过在相关函数调用中加入
seg
参数。这样可以确保在进行前向和后向传播时正确地传递所有必需的参数,从而避免上述类型错误。具体更改:
self.lm()
方法时添加了缺失的seg
参数。测试情况:
请审查这些更改,感谢您的时间和努力!