| { | |
| "epoch": 2.0, | |
| "max_memory_allocated (GB)": 30.34, | |
| "memory_allocated (GB)": 19.36, | |
| "total_memory_available (GB)": 94.62, | |
| "train_loss": 2.0642123500715774, | |
| "train_runtime": 549.193, | |
| "train_samples_per_second": 97.186, | |
| "train_steps_per_second": 1.517 | |
| } |