Train MBART Model: forward() got an unexpected keyword argument 'label_mask'

[Anh-Dung Vo]

Scenario:

• Run: $ ./examples/train_mbart_model.sh

Error:
TypeError: forward() got an unexpected keyword argument 'label_mask'

I'm looking forward to your reply.
Thanks!

[Raj Dabre]

Hi,

Are you using the version of transformers provided with this toolkit, or did you install it using "pip install transformers?"

If you used the latter, then you will get this error.

To fix this:

1. pip uninstall transformers
2. cd yanmtt/transformers
3. python setup.py install

[Anh-Dung Vo]

Dear Dr. @prajdabre ,

Your answer solved my issue perfectly.

Thank you!