train_results.json

{
    "epoch": 3.0,
    "max_memory_allocated (GB)": 87.45,
    "memory_allocated (GB)": 58.84,
    "total_flos": 4.836954571265802e+18,
    "total_memory_available (GB)": 94.62,
    "train_loss": 0.9009929867578731,
    "train_runtime": 19479.8403,
    "train_samples_per_second": 21.776,
    "train_steps_per_second": 1.361
}