Running it purely on CPU gives me following error:
RuntimeError: $ Torch: not enough memory: you tried to allocate 1GB. Buy new RAM! at /pytorch/aten/src/TH/THGeneral.c:218
Could it be possible that my network is itself too big? Considering vocab size is 1.3 Million as stated above, can someone do an analysis of the memory required?