pxyu / pretraining-clir Goto Github PK

Hi, I am getting the following error when running train.py

Traceback (most recent call last):
File "D:/py/CLIR/Pretraining-CLIR-master/train.py", line 156, in
main(params)
File "D:/py/CLIR/Pretraining-CLIR-master/train.py", line 135, in main
trainer.qlm_step(lang_pair, params.lambda_qlm)
File "D:\py\CLIR\Pretraining-CLIR-master\trainer.py", line 340, in qlm_step
outputs = self.model(inputs)
File "D:\anaconda3\envs\CLIR\lib\site-packages\torch\nn\modules\module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "D:\py\CLIR\Pretraining-CLIR-master\BertForXLRetrieval.py", line 46, in forward
return self.forward_mlm(**kwargs)
File "D:\py\CLIR\Pretraining-CLIR-master\BertForXLRetrieval.py", line 87, in forward_mlm
output_hidden_states=output_hidden_states,
File "D:\anaconda3\envs\CLIR\lib\site-packages\torch\nn\modules\module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
TypeError: forward() got an unexpected keyword argument 'output_attentions'

Any idea how to solve this error? Thanks.