train_results.json

{
    "epoch": 0.4,
    "learning_rate": 6.000106106539628e-06,
    "train_loss": 4.860154559897242,
    "train_runtime": 1697.8729,
    "train_samples": 2366765,
    "train_samples_per_second": 557.573,
    "train_steps_per_second": 1.089
}