RuntimeError: CUDA error: device-side assert triggered

[wxr997]

File "D:\Project\Python_project\MyProject\dcp-master\main.py", line 214, in train_one_epoch
total_loss += loss.item() * batch_size
^^^^^^^^^^^
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.
Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.

Hello, I would like to ask what is the reason for the above problem when running the source code? It previously appeared when running to the 70th epoch.