all_results.json

{
    "epoch": 3.0,
    "eval_loss": 3.6610631942749023,
    "eval_mse": 3.6610634619705245,
    "eval_runtime": 54.3093,
    "eval_samples": 96338,
    "eval_samples_per_second": 1773.877,
    "eval_steps_per_second": 221.748,
    "total_flos": 4.283504864539085e+16,
    "train_loss": 3.5571271605219925,
    "train_runtime": 3169.0778,
    "train_samples": 867042,
    "train_samples_per_second": 820.783,
    "train_steps_per_second": 12.825
}