You are training a computer vision model on a GPU-powered virtual machine in Compute Engine to classify government ID types from images. The model uses SGD as the optimizer, with an image shape of 224x224, a batch size of 64, and is set to run for 10 epochs with a verbose of 2. During training, you encounter an 'Out of Memory (OOM)' error. Considering the need to maintain model accuracy while resolving the OOM error, which of the following adjustments should you prioritize? (Choose one correct option)