运行gnn模型 用cnn encoder 报错

[LeiPK]

python train_demo.py
--val val_wiki
--model snail
--encoder cnn
--trainN 10
--N 5
--K 1
--val_step 1000

模型报错如下：
../aten/src/ATen/native/cuda/Indexing.cu:1141: indexSelectLargeIndex: block: [18,0,0], thread: [123,0,0] Assertion srcIndex < srcSelectDimSize failed.
../aten/src/ATen/native/cuda/Indexing.cu:1141: indexSelectLargeIndex: block: [18,0,0], thread: [124,0,0] Assertion srcIndex < srcSelectDimSize failed.
../aten/src/ATen/native/cuda/Indexing.cu:1141: indexSelectLargeIndex: block: [18,0,0], thread: [125,0,0] Assertion srcIndex < srcSelectDimSize failed.
../aten/src/ATen/native/cuda/Indexing.cu:1141: indexSelectLargeIndex: block: [18,0,0], thread: [126,0,0] Assertion srcIndex < srcSelectDimSize failed.
../aten/src/ATen/native/cuda/Indexing.cu:1141: indexSelectLargeIndex: block: [18,0,0], thread: [127,0,0] Assertion srcIndex < srcSelectDimSize failed.
run2.0.sh: line 27: 34102 Segmentation fault python train_demo.py --val val_wiki --model snail --encoder cnn --trainN 10 --N 5 --K 1 --val_step 1000