WangYueFt / dcp

346 stars 90 forks source link

RuntimeError: CUDA error: device-side assert triggered #30

Open wxr997 opened 6 months ago

wxr997 commented 6 months ago

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.