all_results.json

{
    "dr_accuracy": 0.9989326375711575,
    "dr_loss": 0.0037797449622303247,
    "dr_runtime": 39.5752,
    "dr_samples_per_second": 639.189,
    "dr_steps_per_second": 2.502,
    "epoch": 20.0,
    "test_accuracy": 0.9457662701189643,
    "test_loss": 0.467970609664917,
    "test_runtime": 9.5073,
    "test_samples_per_second": 601.222,
    "test_steps_per_second": 2.419,
    "train_loss": 0.02967632641231667,
    "train_runtime": 3368.6231,
    "train_samples_per_second": 150.186,
    "train_steps_per_second": 4.696
}