Should set return_dict=True for bert_model in finetuning.ipynb

[wwwennn]

Hi there,

I noticed that this line should be changed to

reps = bert_model(X_example, attention_mask=X_example_mask, return_dict=True)

for the following cells to work.

Thank you,
Wen

[Christopher Potts]

@wwwennn Thanks for the note! This isn't the behavior I see. Can you verify that you are using transformers==4.3.3?

[wwwennn]

You're right. I was running with transformers=3.4. Things work fine after switching to 4.3.3