More than 3 years have passed since last update.

TextGANでAssertion `srcIndex < srcSelectDimSize` failed.がでた場合

Posted at 2020-11-23

TextGANのgithub
https://github.com/williamSYSU/TextGAN-PyTorch
独自のデータセットに関してSeqGANを適用したら下記のエラー。

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

gpuでの実行をやめてcpuで実行したら以下のエラー。

IndexError: index out of range in self

<解決方法>
/TextGAN-PyTorch/dataset/
にあるXX_iw.dictとXX_wi.txtを削除。
(XXはデータセットの名前(ex..image_coco.txt->image_coco_wi_dict.txt))