运行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