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

[nnx] try casting integers keys in State.replace_by_pure_dict #4317

Merged
merged 1 commit into from
Oct 22, 2024

Conversation

cgarciae
Copy link
Collaborator

What does this PR do?

Tries to cast keys to integerts in State.replace_by_pure_dict to reverse the casting done by State. to_pure_dict.

Fixes #4308.

@copybara-service copybara-service bot merged commit 8360b7c into main Oct 22, 2024
18 of 19 checks passed
@copybara-service copybara-service bot deleted the improve-state-loading branch October 22, 2024 12:49
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.

Issue on restoring NNX state from Orbax checkpoint as pure dict
2 participants