all_results.json

{
    "test_loss": 1.2583799362182617,
    "test_runtime": 10.6119,
    "test_samples_per_second": 35.526,
    "test_steps_per_second": 5.937
}