pytorch-dual-learning
pytorch-dual-learning copied to clipboard
when I run dual.py and the step is around 1000, there is error "RuntimeError: cuda runtime error (2) : out of memory at /pytorch/torch/lib/THC/generic/THCStorage.cu:58"
I run dual.py with GPU 1080Ti and the memory size is 12G. The corpus sizes of nmt and lm are both no more than 50M. But when the step of dual is around 1000, there is error:
Traceback (most recent call last):
File "/ghome/weiyt/pytorch-dual-learning/dual.py", line 202, in