all_results.json

{
    "epoch": 0.4,
    "learning_rate": 5.9914530099610235e-06,
    "train_loss": 4.845045407784673,
    "train_runtime": 1698.7001,
    "train_samples": 2371626,
    "train_samples_per_second": 558.507,
    "train_steps_per_second": 1.091
}