all_results.json

{
    "epoch": 4.0,
    "test_loss": 0.5490162372589111,
    "test_runtime": 869.0719,
    "test_samples_per_second": 2.166,
    "test_steps_per_second": 0.272,
    "train_loss": 0.7414002821479045,
    "train_runtime": 7813.3494,
    "train_samples_per_second": 7.706,
    "train_steps_per_second": 1.926
}