I have been trying to use BertForSequenceClassification for document classification in Docker but while generating the logits using this line outputs = bert_model(**inputs) it just stops. Can anyone suggest me a solution for this or any alternate approach.
Related topics
| Topic | Replies | Views | Activity | |
|---|---|---|---|---|
| Cross-validation for BERT models | 0 | 996 | December 8, 2020 | |
| Further Pretrain Basic BERT for sequence classification | 4 | 1829 | October 9, 2020 | |
| What FineTuning can be done with a available models | 4 | 448 | February 23, 2021 | |
| BertForSequenceClassification finetune training loss and accuracy have some problem | 0 | 876 | December 14, 2020 | |
| Looking for tool class to do predictions | 3 | 564 | October 9, 2020 |