train_results.json

{
    "epoch": 0.4,
    "learning_rate": 6.006595928973583e-06,
    "train_loss": 4.8374716447724815,
    "train_runtime": 1702.9085,
    "train_samples": 2362041,
    "train_samples_per_second": 555.022,
    "train_steps_per_second": 1.084
}