train_results.json

{
    "all_params": 224673024,
    "epoch": 3.0,
    "memory_footprint": 898692096,
    "total_flos": 2506179136462848.0,
    "train_loss": 1.8666806372729214,
    "train_runtime": 383.5951,
    "train_samples_per_second": 6.874,
    "train_steps_per_second": 0.86,
    "trainable_params": 1769472,
    "trainable_params_percent": 0.7875765272113843
}