How to handle RuntimeError: CUDA error: an illegal memory access was encountered?

Question

I am running some code with pytorch 1.7 and cuda 11.0 on my gpu enabled machine. I have a few classes that contain the code for a few different neural networks. My first class r