all_results.json

{
    "dr_accuracy": 0.9156,
    "dr_loss": 0.2775183916091919,
    "dr_runtime": 17.6363,
    "dr_samples_per_second": 2409.803,
    "dr_steps_per_second": 9.469,
    "epoch": 300.0,
    "test_accuracy": 0.8328,
    "test_loss": 0.5651447772979736,
    "test_runtime": 8.6051,
    "test_samples_per_second": 1162.104,
    "test_steps_per_second": 4.648,
    "train_loss": 1.2348953081155802,
    "train_runtime": 17779.0058,
    "train_samples_per_second": 717.138,
    "train_steps_per_second": 5.619
}