)
THCudaCheck FAIL file=/pytorch/aten/src/THC/THCTensorRandom.cu line=25 error=2 : out of memory
Traceback (most recent call last):
File “train.py”, line 110, in 
main()
File “train.py”, line 56, in main
dream_model.cuda()
File “/mnt/home/dreamdd/newDream/DreamHU/venv/lib/python2.7/site-packages/torch/nn/modules/module.py”, line 249, in cuda
return self._apply(lambda t: t.cuda(device))
File “/mnt/home/dreamdd/newDream/DreamHU/venv/lib/python2.7/site-packages/torch/nn/modules/module.py”, line 176, in _apply
module._apply(fn)
File “/mnt/home/dreamdd/newDream/DreamHU/venv/lib/python2.7/site-packages/torch/nn/modules/module.py”, line 176, in _apply
module._apply(fn)
File “/mnt/home/dreamdd/newDream/DreamHU/venv/lib/python2.7/site-packages/torch/nn/modules/module.py”, line 176, in _apply
module._apply(fn)
File “/mnt/home/dreamdd/newDream/DreamHU/venv/lib/python2.7/site-packages/torch/nn/modules/module.py”, line 182, in _apply
param.data = fn(param.data)
File “/mnt/home/dreamdd/newDream/DreamHU/venv/lib/python2.7/site-packages/torch/nn/modules/module.py”, line 249, in 
return self._apply(lambda t: t.cuda(device))
RuntimeError: cuda runtime error (2) : out of memory at /pytorch/aten/src/THC/THCTensorRandom.cu:25