Image by Author

Pytorch Conv2d Weights Explained

Understanding weights dimension, visualization, number of parameters and the infamous size mismatch

One of the most common problems I have found in my journey with Pytorch is the size mismatch error when uploading weights to my models. As you know, Pytorch does not save the computational graph of your model when you save the model weights…