all_results.json

{
    "dr_accuracy": 0.9971764705882353,
    "dr_loss": 0.014752509072422981,
    "dr_runtime": 122.6318,
    "dr_samples_per_second": 346.566,
    "dr_steps_per_second": 1.362,
    "epoch": 300.0,
    "test_accuracy": 0.9301,
    "test_loss": 0.25753843784332275,
    "test_runtime": 30.1063,
    "test_samples_per_second": 332.157,
    "test_steps_per_second": 1.329,
    "train_loss": 0.2884208381307256,
    "train_runtime": 121518.863,
    "train_samples_per_second": 104.922,
    "train_steps_per_second": 0.822
}