all_results.json

{
    "epoch": 3.0,
    "test_loss": 0.005226554349064827,
    "test_overall_accuracy": 0.9851129807294873,
    "test_overall_f1": 0.9270450073152169,
    "test_overall_precision": 0.9281906912835416,
    "test_overall_recall": 0.9259021481405626,
    "test_runtime": 2690.9722,
    "test_samples_per_second": 150.748,
    "test_steps_per_second": 4.711
}