How to deal with memory issues in GPU and CPU when running Keras Tensorflow Script

Question

I am running a script that basically processes 39937 images with 224x224x3 shape using a RESNET50 CNN in a fine-tuning fashion. The code is below:

import os
i