all_results.json

{
    "epoch": 3.0,
    "eval_loss": 0.5030496716499329,
    "eval_mse": 0.5030496716499329,
    "eval_runtime": 32.9075,
    "eval_samples": 4009,
    "eval_samples_per_second": 121.826,
    "eval_steps_per_second": 15.255,
    "train_loss": 0.7615933273777817,
    "train_runtime": 2470.73,
    "train_samples": 31069,
    "train_samples_per_second": 37.724,
    "train_steps_per_second": 0.294
}