all_results.json

{
    "dr_accuracy": 0.9990512333965844,
    "dr_loss": 0.003322172211483121,
    "dr_runtime": 38.9939,
    "dr_samples_per_second": 648.718,
    "dr_steps_per_second": 2.539,
    "epoch": 20.0,
    "test_accuracy": 0.9522393282015396,
    "test_loss": 0.43564239144325256,
    "test_runtime": 9.413,
    "test_samples_per_second": 607.247,
    "test_steps_per_second": 2.443,
    "train_loss": 0.03330213655864846,
    "train_runtime": 3337.005,
    "train_samples_per_second": 151.609,
    "train_steps_per_second": 4.741
}