Figure 4. Training and validation losses of the final model over 300,000 iterations