epoch_5/train_results.json

{
    "epoch": 1.0,
    "total_flos": 14669778124800.0,
    "train_loss": 0.14801818221553484,
    "train_runtime": 3072.7463,
    "train_samples_per_second": 2.864,
    "train_steps_per_second": 1.432
}