{ "epoch": 3.0, "max_memory_allocated (GB)": 2.1, "memory_allocated (GB)": 1.46, "total_flos": 6.324139790696448e+19, "total_memory_available (GB)": 126.62, "train_loss": NaN, "train_runtime": 4358.6451, "train_samples_per_second": 187.213, "train_steps_per_second": 23.402 }