The problem of out of memory

Please for help.

I trained the model, which was constructed in Pytorch.
But there is a error. It shows in the figure.

The environment is:
Ubuntu 16.04
Python 3.6
Pytorch 1.1.0

My GPU is GeForce GTX 1080.

The first thing you should do is set the mini-batch size to a smaller number, e.g a half of your current size.
This error occurs when your network need more memory than the available amount in your system.
Hope this help!

1 Like

Thank you for your suggestion
I will test it