{ "epoch": 2.0, "train_loss": 0.2957918119430542, "train_runtime": 128.4347, "train_samples_per_second": 24.915, "train_steps_per_second": 0.389 }