terminate called after throwing an instance of 'c10::CUDAError' what(): CUDA error: device-side assert triggered

[andyoung009]

Hi, i am interested in this project. When i run the code, the error comes like following:
../aten/src/ATen/native/cuda/ScatterGatherKernel.cu:111: operator(): block: [310285,0,0], thread: [127,0,0] Assertion idx_dim >= 0 && idx_dim < index_size && "index out of bounds" failed.
terminate called after throwing an instance of 'c10::CUDAError'
what(): CUDA error: device-side assert triggered
Exception raised from createEvent at ../aten/src/ATen/cuda/CUDAEvent.h:174 (most recent call first):
frame #0: c10::Error::Error(c10::SourceLocation, std::string) + 0x42 (0x7f0f18be27d2 in
../miniconda3/envs/ATS/lib/python3.8/site-packages/torch/lib/libc10.so)
frame #1: + 0x267df7a (0x7f0f6bc01f7a in ../miniconda3/envs/ATS/lib/python3.8/site-packages/torch/lib/libtorch_cuda_cpp.so)
frame #2: + 0x300568 (0x7f0fcdffa568 in ../miniconda3/envs/ATS/lib/python3.8/site-packages/torch/lib/libtorch_python.so)
frame #3: c10::TensorImpl::release_resources() + 0x175 (0x7f0f18bcb005 in ../miniconda3/envs/ATS/lib/python3.8/site-packages/torch/lib/libc10.so)
frame #4: + 0x1ee569 (0x7f0fcdee8569 in ../miniconda3/envs/ATS/lib/python3.8/site-packages/torch/lib/libtorch_python.so)
frame #5: + 0x4d9c78 (0x7f0fce1d3c78 in ../miniconda3/envs/ATS/lib/python3.8/site-packages/torch/lib/libtorch_python.so)
frame #6: THPVariable_subclass_dealloc(_object*) + 0x292 (0x7f0fce1d3f72 in ../miniconda3/envs/ATS/lib/python3.8/site-packages/torch/lib/libtorch_python.so)

frame #34: __libc_start_main + 0xe7 (0x7f0fd0197c87 in /lib/x86_64-linux-gnu/libc.so.6)
Is there anything wrong? I use the code and change the realtive file dir without changing other anything.