29.6x Less Checkpoint Work for PyTorch Model Saving — Exact Reload
PyTorch Forums
29.6x Less Checkpoint Work for PyTorch Model Saving — Exact Reload
29.6x Less Checkpoint Work for PyTorch Model Saving — Exact Reload I’ve been experimenting with a different way to handle persistent model checkpoints when only a small fraction of model state changes between versions. Instead of writing the entire state again for every checkpoint, HKD Checkpoint stores the initial state and then records the active changes required for deterministic continuation. A filesystem benchmark using actual file writes, fsync(), and exact reload: 2,000,000 float32 e...
0 comments
No comments yet.