all_results.json

{
    "epoch": 5.0,
    "train_loss": 0.11400984666692955,
    "train_runtime": 1175.3679,
    "train_samples": 67349,
    "train_samples_per_second": 286.502,
    "train_steps_per_second": 8.955,
    "validation_accuracy": 0.9254587155963303,
    "validation_loss": 0.31086066365242004,
    "validation_runtime": 1.4824,
    "validation_samples": 872,
    "validation_samples_per_second": 588.248,
    "validation_steps_per_second": 18.889
}