My error message:
Traceback (most recent call last):
File “train.py”, line 215, in 
loss_G1.backward()
File “/scratch_net/biwidl207/ligua/anaconda/lib/python2.7/site-packages/torch/tensor.py”, line 93, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph)
File “/scratch_net/biwidl207/ligua/anaconda/lib/python2.7/site-packages/torch/autograd/init.py”, line 90, in backward
allow_unreachable=True)  # allow_unreachable flag
RuntimeError: CUDA error: out of memory