OOM error with GPU


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?


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.)