train_results.json

{
    "epoch": 8.0,
    "max_memory_allocated (GB)": 57.18,
    "memory_allocated (GB)": 50.57,
    "total_flos": 4.89583144415232e+16,
    "total_memory_available (GB)": 94.62,
    "train_loss": 0.057532464606421335,
    "train_runtime": 1666.2328,
    "train_samples_per_second": 52.538,
    "train_steps_per_second": 1.315
}