{ "epoch": 3.0, "max_memory_allocated (GB)": 90.15, "memory_allocated (GB)": 67.67, "total_memory_available (GB)": 94.62, "train_loss": 1.4381873343839031, "train_runtime": 1086.3269, "train_samples_per_second": 16.197, "train_steps_per_second": 1.015 }