TypeError: forward() got an unexpected keyword argument 'masked_lm_labels'

utterworks / fast-bert

Super easy library for BERT based NLP models

Apache License 2.0

1.86k stars 341 forks source link

TypeError: forward() got an unexpected keyword argument 'masked_lm_labels' #310

Open FirstGalacticEmpire opened 2 years ago

FirstGalacticEmpire commented 2 years ago

learner.fit(epochs=1, r=6e-5, validate=True, # Evaluate the model after each epoch schedule_type="warmup_cosine", optimizer_type="lamb") Hi, following the official tutorial ("Language Model Fine-tuning) , i get the following error presented in screenshot, while running .fit function.