Sync the shapes of the batches in multi you training to make the times more similar and save total time, as no machine waits.