whene i try to train my neural network i meet this error:
torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File “/home/imene/anaconda3/envs/pose/lib/python3.6/site-packages/torch/nn/modules/conv.py”, line 301, in forward
self.padding, self.dilation, self.groups)
RuntimeError: CUDA error: out of memory
I tried to put batch size to 1 but that does not work,
I have an i7 RAM = 12G, GPU = [GeForce GTX 750 Ti] (rev a2) 2G
can any one give me solution because i am blocked since tow weeks
tanks.