all_results.json

{
    "epoch": 0.0,
    "train_loss": 0.662235689163208,
    "train_runtime": 78.1728,
    "train_samples_per_second": 5.117,
    "train_steps_per_second": 0.128
}