abhishekkrthakur / bert-sentiment

MIT License
268 stars 103 forks source link

line 15 model.py #19

Open alvynabranches opened 2 years ago

alvynabranches commented 2 years ago

TypeError: dropout(): argument 'input' (position 1) must be Tensor, not str

arush414 commented 1 year ago

Hey just add an extra argument in the self.bert argument ie: return_dictionary=False