train_results.json

{
    "epoch": 2.0,
    "tflops": 490.2053045046936,
    "token/s": 1736.022262942593,
    "train_loss": 0.035910474946516446,
    "train_runtime": 31598.4886,
    "train_samples_per_second": 6.603,
    "train_steps_per_second": 0.026
}