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.

Sign up for GitHub

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

Jump to bottom

[Schedule] Support partial checkpointing #55

Merged
merged 4 commits into from
Feb 13, 2023
Merged

[Schedule] Support partial checkpointing #55

merged 4 commits into from
Feb 13, 2023

Conversation

chhzh123
Copy link
Contributor

Description

This PR adds support to partial checkpointing, i.e., only checkpoint a subgraph. Users can use .find() primitive to obtain the required subgraph and then pass it to .checkpoint() primitive. A new checkpointed module will be attached to the original parent module.

Checklist

  • PR's title starts with a category (e.g. [Bugfix], [Model], [Tutorial], etc)
  • Changes are complete (i.e. I finished coding on this PR)
  • All changes have test coverage
  • Code is well-documented

@chhzh123 chhzh123 requested a review from comaniac February 13, 2023 17:13
@chhzh123 chhzh123 mentioned this pull request Feb 13, 2023
Open
15 tasks
comaniac
comaniac approved these changes Feb 13, 2023
View reviewed changes
Copy link
Contributor

@comaniac comaniac left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. Just need to improve comments.

@chhzh123 chhzh123 merged commit 3ef4139 into awslabs:main Feb 13, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@comaniac comaniac comaniac approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@chhzh123 @comaniac