pytorch-lightning
Sharded state dicts save correctly when `save_weights_only=True`
#19524
Merged
Overview
Commits
15
Changes
View On GitHub

Sharded state dicts save correctly when `save_weights_only=True` #19524

awaelchli merged 15 commits into Lightning-AI:master from dimitri-voytan:fix/fsdp-checkpoint
dimitri-voytan
dvoytan-spark test saving sharded weights only
2e54ca5f
dvoytan-spark add fix for weights only
3b35e2cb
dimitri-voytan Merge branch 'Lightning-AI:master' into fix/fsdp-checkpoint
9f5db5a9
github-actions github-actions added pl
pre-commit-ci[bot] [pre-commit.ci] auto fixes from pre-commit.com hooks
99414fca
dvoytan-spark focus test on just full vs sharded
fbc9985c
dvoytan-spark Merge branch 'fix/fsdp-checkpoint' of https://github.com/dimitri-voyt…
80efc9cb
pre-commit-ci[bot] [pre-commit.ci] auto fixes from pre-commit.com hooks
1fe2038c
dimitri-voytan Merge branch 'master' into fix/fsdp-checkpoint
108ac6b4
dimitri-voytan
dimitri-voytan dimitri-voytan marked this pull request as ready for review 2 years ago
dimitri-voytan dimitri-voytan requested a review from awaelchli awaelchli 2 years ago
dimitri-voytan dimitri-voytan requested a review from carmocca carmocca 2 years ago
dimitri-voytan dimitri-voytan requested a review from justusschock justusschock 2 years ago
dimitri-voytan dimitri-voytan requested a review from Borda Borda 2 years ago
dimitri-voytan dimitri-voytan requested a review from williamFalcon williamFalcon 2 years ago
awaelchli awaelchli added strategy: fsdp
awaelchli awaelchli added bug
awaelchli awaelchli added community
awaelchli awaelchli added this to the 2.2.x milestone 2 years ago
awaelchli awaelchli assigned awaelchli awaelchli 2 years ago
awaelchli reuse existing test
89a80c17
awaelchli update chlog
67efd9c6
mergify mergify added has conflicts
awaelchli
awaelchli approved these changes on 2024-03-08
awaelchli Merge branch 'master' into fix/fsdp-checkpoint
b27e620d
mergify mergify removed has conflicts
awaelchli fix precommit
c3610750
Borda
Borda approved these changes on 2024-03-08
mergify mergify added ready to be merged
carmocca
carmocca approved these changes on 2024-03-08
awaelchli oops
53dd3225
Borda Merge branch 'master' into fix/fsdp-checkpoint
d132d9db
awaelchli Merge branch 'master' into fix/fsdp-checkpoint
81e693ce
awaelchli awaelchli merged b3275e05 into master 2 years ago
jthomy

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
bug ready to be merged strategy: fsdp community pl
Milestone
2.2.x
Terms Of Use Privacy Policy Refund Policy Impressum