Skip to content

[KGE] Add checkpointing utils#342

Merged
nshah-sc merged 3 commits intomainfrom
nshah/add-ckpt-code
Oct 2, 2025
Merged

[KGE] Add checkpointing utils#342
nshah-sc merged 3 commits intomainfrom
nshah/add-ckpt-code

Conversation

@nshah-sc
Copy link
Copy Markdown
Collaborator

Add utils to save/load distributed / sharded checkpoints for KGE, using DCP utils.

Where is the documentation for this feature?: N/A

Did you add automated tests or write a test plan?

Updated Changelog.md? NO

Ready for code review?: YES

Comment thread python/gigl/experimental/knowledge_graph_embedding/lib/checkpoint.py Outdated
Copy link
Copy Markdown
Collaborator

@svij-sc svij-sc left a comment

Choose a reason for hiding this comment

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

Few comments, otherwise LGTM

@nshah-sc nshah-sc marked this pull request as ready for review October 1, 2025 23:27
@nshah-sc nshah-sc added this pull request to the merge queue Oct 1, 2025
Merged via the queue into main with commit 5b15da4 Oct 2, 2025
4 checks passed
@nshah-sc nshah-sc deleted the nshah/add-ckpt-code branch October 2, 2025 00:47
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.

3 participants