RuntimeError: CUDA out of memory

MichalBusta / E2E-MLT

E2E-MLT - an Unconstrained End-to-End Method for Multi-Language Scene Text

MIT License

291 stars 84 forks source link

RuntimeError: CUDA out of memory #37

Closed shin7 closed 5 years ago

shin7 commented 5 years ago

Hello, I have met a problem when training the model as follow:

RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 983.69 MiB total capacity; 209.53 MiB already allocated; 69.56 MiB free; 6.47 MiB cached)

could you give me any solutions? Thank you!

kei6 commented 5 years ago

It seems your device is not strong enough to train this model. 1G of GPU is really small. I'm using 16Gb GPU and this model uses its all.