{ "epoch": 3.0, "max_memory_allocated (GB)": 87.45, "memory_allocated (GB)": 58.84, "total_flos": 4.224438461446226e+17, "total_memory_available (GB)": 94.62, "train_loss": 0.8718739162030859, "train_runtime": 1768.6066, "train_samples_per_second": 21.704, "train_steps_per_second": 1.358 }