I originally planned to train for 20 epochs of 5440 steps, but the performance of the 9th epoch was good enough.
