You are training an ML model on a large dataset using Tensor Processing Units (TPUs) to accelerate the training process. Despite this, you notice that the training is taking longer than expected. Upon investigation, you discover that the TPU is not being fully utilized, with its capacity not reaching optimal levels. What adjustment should you make to improve TPU utilization and speed up the training process?