vdcmclou:

After further investigating I think there is something wrong with my overall structure but I’m not sure as to what it is.


That might be the case and you could compare and debug your model architecture using working implementations such as the torchvision.models.resnet**.