forward() got an unexpected keyword argument 'device'

[basti97]

Hello,

thanks for sharing your code.
I am new to BERT and finetuning and I tried your code with costumn data.

I recieved following error on running DocumentBertMaxPool/DocumentBertLinear:

_` batch_predictions = self.bert_doc_classification(batch_document_tensors,
185 batch_document_sequence_lengths,
--> 186 freeze_bert=self.args['freeze_bert'], device=self.args['device'])
187
188 batch_correct_output = correct_output[i:i + self.args['batch_size']].to(device=self.args['device'])

/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py in call(self, *input, **kwargs)
530 result = self._slow_forward(*input, **kwargs)
531 else:
--> 532 result = self.forward(*input, **kwargs)
533 for hook in self._forward_hooks.values():
534 hook_result = hook(self, input, result)

TypeError: forward() got an unexpected keyword argument 'device'_
`

I think the reason for this problem is that there is missing the 'freeze_bert'-argument in the forward-method of the model architecture (document_bert_architectures.py).

Thanks for response.

[zeinramadan]

I am getting a similar error with the forward function, did you manage to figure out what was the issue?