code/train_results.json

{
    "epoch": 1.0,
    "train_loss": 0.3343511177943303,
    "train_runtime": 232.7119,
    "train_samples_per_second": 4.443,
    "train_steps_per_second": 0.279
}