all_results.json

{
    "dr_accuracy": 0.9128705882352941,
    "dr_loss": 0.28069740533828735,
    "dr_runtime": 17.764,
    "dr_samples_per_second": 2392.476,
    "dr_steps_per_second": 9.401,
    "epoch": 300.0,
    "test_accuracy": 0.8306,
    "test_loss": 0.5670732259750366,
    "test_runtime": 5.7136,
    "test_samples_per_second": 1750.217,
    "test_steps_per_second": 7.001,
    "train_loss": 1.232413830227322,
    "train_runtime": 17888.831,
    "train_samples_per_second": 712.735,
    "train_steps_per_second": 5.584
}