I made a small neural network for image recognition, images in RGB and have dimensions 753x753. Moreover, during training, more than 20GB of memory is used. I fed images thr