I am making a CNN using Pytorch for an image classification problem between people who are wearing face masks and who aren\'t. The images are converted to a 256x256 with 3 c