Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Address deprecation in torch.distributed.checkpoint #6773

Merged
merged 2 commits into from
Mar 20, 2024

Conversation

jonb377
Copy link
Collaborator

@jonb377 jonb377 commented Mar 19, 2024

The save_state_dict and load_state_dict APIs are deprecated, being replaced by save and load APIs. This change updates CheckpointManager and relevant documentation to move away from these deprecated APIs.

Originally reported in #6478

@jonb377 jonb377 requested a review from JackCaoG March 19, 2024 17:31
@jonb377 jonb377 changed the title Address deprecation Address deprecation in torch.distributed.checkpoint Mar 19, 2024
@jonb377 jonb377 force-pushed the jonbolin/deprecation branch from f69fb83 to 3774d8d Compare March 19, 2024 22:04
@jonb377 jonb377 merged commit 73629de into master Mar 20, 2024
19 checks passed
@jonb377 jonb377 deleted the jonbolin/deprecation branch March 20, 2024 16:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants