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