Error when training

Questions
#1

Hello everyone,

I am getting the following error when starting training.

/pytorch/aten/src/THC/THCTensorIndex.cu:272: indexSelectLargeIndex: block: [22,0,0], thread: [127,0,0] Assertion srcIndex < srcSelectDimSize failed.
terminate called after throwing an instance of ‘dmlc::Error’

I think that the graphs are being created correctly. Any idea of what is the source of this error?

Thank you in advance.

#2

Could you provide the versions of your PyTorch and DGL?

Also, it seems that your error is thrown in GPU code. Maybe you can get better error message if you run on CPU again.