Getting an error with vector sizes, while following BERTAbs finetuning tutorial.

CurationCorp / curation-corpus

Code for obtaining the Curation Corpus abstractive text summarisation dataset

Creative Commons Attribution 4.0 International

122 stars 27 forks source link

Getting an error with vector sizes, while following BERTAbs finetuning tutorial. #3

Closed zishanmuzeeb closed 4 years ago

zishanmuzeeb commented 4 years ago

HenryDashwood commented 4 years ago

I've gone through the notebook and found two bugs. The first was a change in the model name on Huggingface's end. The second was a missing training=True parameter that needed to be passed to the collate function. I've fixed them both. Let me know if you keep hitting problems!

zishanmuzeeb commented 4 years ago

I made those changes, These errors are after changing them while computing line 339

HenryDashwood commented 4 years ago

You're getting that when running the same cell as before?