I’m running a kenel that uses pytorch and CNN for mnist digit classification. The model is fairly simple but I’m still getting CUDA OUT  OF MEMORY error.
Here is the link to the kernel :     https://www.kaggle.com/whizzkid/training-best-cnn-model-pytorch 29
I think I’m using pytorch is some wrong way. Because I’ve seen more complex model getting trained on kaggle successfully. I tried same model with keras and it worked as well. Even batch size is 16
Kaggle provides 16GB gpu. That’s a lot to train my model. Still i’m getting the error I don’t know why.
Can anyone tell me what am I doing wrong ?