all_results.json

{
    "epoch": 0.4,
    "learning_rate": 5.9979428323949775e-06,
    "train_loss": 4.861115920608108,
    "train_runtime": 1697.2814,
    "train_samples": 2368021,
    "train_samples_per_second": 558.069,
    "train_steps_per_second": 1.09
}