pytorch/ignite
GitHub ã§èŠãSupport for TorchSnapshot for efficient checkpoint saving and loading
Open
#2,752 opened on 2022幎10æ24æ¥
enhancementhelp wanted
Repository metrics
- Stars
- Â (4,313 stars)
- PR merge metrics
-  (å¹³åããŒãž 18d 23h) (30d 㧠19 merged PRs)
説æ
ð Feature
TorchSnapshot is a performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind. It includes many optimizations to control for memory usage and optimize checkpoint writing for DDP-style workloads over torch.save/torch.load. For more information, please check out the readme: https://github.com/pytorch/torchsnapshot#why-torchsnapshot
This could be a nice addition to Ignite, similar to the existing Checkpoint handler
cc @yifuwang