CUDA out of memory
Tried reducing the batch size to 4 but still…Screen Shot 2022-07-15 at 4.16.10 PM872×446 81.3 KB
2022-07-15T18:25:13Z
What model are you using? It may be too large for 1 GPU. Try with batch size 1
2022-07-16T13:13:58Z
I’m using XLM-RoBERTa-Large model for zero-shot classification.