all_results.json

{
    "epoch": 2.0,
    "train_loss": 1.5770846359804964,
    "train_runtime": 13419.9836,
    "train_samples_per_second": 10.287,
    "train_steps_per_second": 0.08
}