I have more than 50% of the memory on GPU at the moment, I reset the kernel and even reduced the batch size to 1 but I still received the same error as above, or this error RuntimeError: CUDA error: device-side assert triggered