Description
File "predict_multi_scale_multi_loss.py", line 52, in
model.predict_for_regress((test_documents, test_labels))
File "/Multi-Scale-BERT-AES/model_architechure_bert_multi_scale_multi_loss.py", line 72, in predict_for_regress
batch_predictions_word_document = self.bert_regression_by_word_document(batch_document_tensors_word_document, device=self.args['device'])
File "/python3.7/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, **kwargs)
File "/Multi-Scale-BERT-AES/document_bert_architectures.py", line 75, in forward
attention_mask=document_batch[doc_id][:self.bert_batch_size, 2])
File "/python3.7/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, **kwargs)
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)