[PYTHON] When Assertion `srcIndex <srcSelectDimSize` failed. Appears in TextGAN

TextGAN github https://github.com/williamSYSU/TextGAN-PyTorch The following error when applying SeqGAN to my own dataset.

...
Assertion 'srcIndex < srcSelectDimSize' failed
...
...
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`

When I stop executing with gpu and execute with cpu, the following error occurs.

IndexError: index out of range in self

/TextGAN-PyTorch/dataset/ Deleted XX_iw.dict and XX_wi.txt in. (XX is the name of the dataset (ex..image_coco.txt-> image_coco_wi_dict.txt))