prateekjoshi565 / Fine-Tuning-BERT

Apache License 2.0
139 stars 119 forks source link

fix #6 #7

Closed JoKerDii closed 3 years ago

JoKerDii commented 3 years ago

I got ' 'str' object has no attribute 'dim'' by simply running the notebook. I figured it out by looking at this. It seems either specifying 'return_dict=False' or calling 'values()' method could force the bert model to return a tuple. Hope this helps.