all_results.json

{
    "epoch": 5.0,
    "predict_runtime": 3.2345,
    "predict_samples": 1497,
    "predict_samples_per_second": 462.819,
    "predict_steps_per_second": 58.123,
    "train_loss": 1.0310750216723763,
    "train_runtime": 163.2592,
    "train_samples": 5984,
    "train_samples_per_second": 183.267,
    "train_steps_per_second": 5.727
}