CUDA error: device-side assert triggered even after a few forward passes
The notebook for my code is this link: https://colab.research.google.com/drive/1NMS2ki9El-ncOcP4a3dEJNcUg_5zDNeS?usp=sharing
The notebook for my code is this link: https://colab.research.google.com/drive/1NMS2ki9El-ncOcP4a3dEJNcUg_5zDNeS?usp=sharing