train_results.json

{
    "epoch": 3.0,
    "max_memory_allocated (GB)": 2.31,
    "memory_allocated (GB)": 1.67,
    "total_flos": 6.324139790696448e+19,
    "total_memory_available (GB)": 126.62,
    "train_loss": NaN,
    "train_runtime": 3737.7146,
    "train_samples_per_second": 218.314,
    "train_steps_per_second": 27.289
}