{
    "epoch": 5.0,
    "train_loss": 0.0,
    "train_runtime": 0.2484,
    "train_samples_per_second": 20812.907,
    "train_steps_per_second": 2616.71
}