CUDA out of memory

🤗Transformers
1

Tried reducing the batch size to 4 but still…

Screen Shot 2022-07-15 at 4.16.10 PM
Screen Shot 2022-07-15 at 4.16.10 PM872×446 81.3 KB

2

What model are you using? It may be too large for 1 GPU. Try with batch size 1

3

I’m using XLM-RoBERTa-Large model for zero-shot classification.