train_model, must be Tensor, not str

[DanielEdu]

Hola!

he probado el notebook pero me da error al momento de entrenar el modelo

TypeError: dropout(): argument 'input' (position 1) must be Tensor, not str
probablemente actualizaron parámetros en el transformador, pero no logro identificarlo

[jcsolares]

Hola, puedes corregirlo agregando el parametro return_dict=False a la clase asi:
cls_output = self.bert(
input_ids = input_ids,
attention_mask = attention_mask,
return_dict=False
)