{
    "epoch": 3.0,
    "test_accuracy": 0.6684,
    "test_f1_macro": 0.6172965996785956,
    "test_f1_micro": 0.6684,
    "test_loss": 0.9713044762611389,
    "test_runtime": 4.7688,
    "test_samples_per_second": 1048.472,
    "test_steps_per_second": 16.566
}