CUDA error: device-side assert triggered

[cyxhello]

Hello,
Thanks for opening your source code. It's very nice works.

when i running the code of sentgcn.py， there are some errors following:
Traceback (most recent call last):
File "train_sentgcn.py", line 195, in
preds = model(images1, images2, stop_id=embs.word_indexer.index_of('.'))
File "/home/cyx/yes/envs/avalanche/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "/home/cyx/MeRG/mekg/sentgcn.py", line 286, in forward
x_t = logit.argmax(dim=1)
RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

Could you tell me how solve the issue? very thanks!