train_results.json

{
    "all_params": 60624768,
    "epoch": 3.0,
    "memory_footprint": 242499072,
    "total_flos": 357853398958080.0,
    "train_loss": 2.845433976433494,
    "train_runtime": 116.68,
    "train_samples_per_second": 22.6,
    "train_steps_per_second": 2.828,
    "trainable_params": 118144,
    "trainable_params_percent": 0.19487744678874483
}