train_output.json

[
  53962,
  0.6364423901274552,
  {
    "train_runtime": 6417.4177,
    "train_samples_per_second": 8.409,
    "train_steps_per_second": 8.409,
    "train_loss": 0.6364423901274552,
    "epoch": 2.0
  }
]