RuntimeError: CUDA error: an illegal memory access was encountered
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.
when im training with my own colab note book https://colab.research.google.com/drive/15wOg4Nl99y2CMqNc9gwl82i3mypJs60D?usp=sharing