basti97

Target size (torch.Size([10])) must be the same as input size (torch.Size([10, 6]))

2

Another error occurs while running DocumentBertLSTM: _ValueError Traceback (most recent call last) in () 7 8 model = BertForDocumentClassification(args=args) ----> 9 model.fit((train_documents, train_labels), (dev_documents,dev_labels)) 3 frames /usr/local/lib/python3.7/dist-packages/bert_document_classification/document_bert.py in fit(self, train,...

forward() got an unexpected keyword argument 'device'

1

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...