[PYTHON] Lorsque l'assertion `srcIndex <srcSelectDimSize` a échoué. Apparaît dans TextGAN

TextGAN github https://github.com/williamSYSU/TextGAN-PyTorch L'erreur suivante lors de l'application de SeqGAN pour mon propre ensemble de données.

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

Lorsque j'arrête de m'exécuter avec gpu et que j'exécute avec cpu, l'erreur suivante se produit.

IndexError: index out of range in self

/TextGAN-PyTorch/dataset/ Supprimé XX_iw.dict et XX_wi.txt dans. (XX est le nom de l'ensemble de données (ex..image_coco.txt-> image_coco_wi_dict.txt))