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 | |
---|---|---|---|---|
Unable to Run Sentence Transformer Text embedding in Docker | 1 | 271 | January 7, 2025 | |
Bert for audio classification | 0 | 1144 | April 25, 2022 | |
Upload Bert style classifier with mean pooling | 0 | 345 | August 25, 2023 | |
Implementation difference between Bert and Roberta ForSequenceClassification? | 0 | 556 | June 24, 2021 | |
Why BertForSequenceClassification performs worse than BertModel+Linear? | 0 | 334 | February 24, 2022 |