train_results.json

{
    "all_params": 224673024,
    "epoch": 3.0,
    "memory_footprint": 898692096,
    "total_flos": 2506179136462848.0,
    "train_loss": 1.8389299175956033,
    "train_runtime": 453.2247,
    "train_samples_per_second": 5.818,
    "train_steps_per_second": 0.728,
    "trainable_params": 1769472,
    "trainable_params_percent": 0.7875765272113843
}