JuanFMontesinos:

etting outnof memory it means you did a mistake in the code.


I don’t totally understand what you said. But what you said is correct.
I changed my optimizer to SGD. But my GPU memory is imbalanced I think just like you said. But I don’t know how to avoid that.
Second thing: My code doesn’t even run one iteration
Third Thing: Is the memory not being freed? I think No. Below is my screenshot.! (Top is the Nvidia Memory)(Below is my Error). Please help me and guide me what to do

Screen Shot 2018-11-20 at 4.04.14 PM.png1684×1510 334 KB