TextGANでAssertion `srcIndex < srcSelectDimSize` failed.がでた場合
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))
Author And Source
この問題について(TextGANでAssertion `srcIndex < srcSelectDimSize` failed.がでた場合), 我々は、より多くの情報をここで見つけました https://qiita.com/sk161717/items/5d1404fffeb9fb7b9780著者帰属:元の著者の情報は、元のURLに含まれています。著作権は原作者に属する。
Content is automatically searched and collected through network algorithms . If there is a violation . Please contact us . We will adjust (correct author information ,or delete content ) as soon as possible .