You are in the midst of training an ML model on a sizable dataset, and you are utilizing a TPU (Tensor Processing Unit) to accelerate the training process. However, you've noticed that the training is proceeding slower than expected, and upon investigation, you've determined that the TPU is not fully utilizing its capacity. What actions should you take to address this issue?