pytorch-lightning
(3/n) Support 2D Parallelism - Efficient loading of full-state checkpoints
#19870
Merged
Merged
(3/n) Support 2D Parallelism - Efficient loading of full-state checkpoints #19870
lantiga
merged
7 commits into
master
from
feature/tp-full-ckpt-load
Base automatically changed from
examples/tp-ckpt
to
master
1 year ago
memory-optimized loading of full checkpoints into dist model
1a0887ec
simplify
d1e5bd15
handle buffers
63719999
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&s=60)
[pre-commit.ci] auto fixes from pre-commit.com hooks
91664a1d
handle strict loading, buffers, and add test
4f861f64
[pre-commit.ci] auto fixes from pre-commit.com hooks
fcb7fdba
chlog
bd2843f6
awaelchli
changed the title
awaelchli
commented
on 2024-05-15
justusschock
approved these changes
on 2024-05-15
lantiga
approved these changes
on 2024-05-15
Reviewers
Assignees
No one assigned
Labels
ready to be merged
refactor
fabric
performance
pl
Milestone
Login to write a write a comment.
Login via GitHub