{
    "epoch": 94.0,
    "eval_explained_variance": 0.4649323523044586,
    "eval_kl_divergence": 0.3295058012008667,
    "eval_loss": 0.46722450852394104,
    "eval_mae": 0.11545699089765549,
    "eval_rmse": 0.15503521263599396,
    "eval_runtime": 63.8487,
    "eval_samples_per_second": 36.915,
    "eval_steps_per_second": 2.318,
    "learning_rate": 1.0000000000000002e-07
}