OOM error with GPU

DeepPavlov Library
1

Hello!

I use bert model for classification (rubert ot multilanguage as BertPreprocessor).
I have OOM error (cpu and gpu).
How can I manage parameters for learning model?

Thanks.

2

Hi @AlexModestov,

Could you send exact code, commands and logs with the out of memory error? And please share your system specifications (OS, RAM, GPU, etc.)