@emross3371 This can be explained. If you have very small images then during the later layers, you dont have enough pixels left to do complete convolution and therefore shape is reduced to 1. Every image processing model should accept a particular shape of the image. If 224 is working fine, then resize the 28x28x3 to 224x224x3 using torch Resize
https://pytorch.org/vision/stable/transforms.html#torchvision.transforms.Resize