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] fix Accuracy on eager mode #4065

Merged
merged 1 commit into from
Jul 9, 2024
Merged

[nnx] fix Accuracy on eager mode #4065

merged 1 commit into from
Jul 9, 2024

Conversation

cgarciae
Copy link
Collaborator

@cgarciae cgarciae commented Jul 9, 2024

What does this PR do?

Casts Accuracy labels to jnp.int32 to avoid issues with eager mode.

@copybara-service copybara-service bot merged commit 0da4f31 into main Jul 9, 2024
18 checks passed
@copybara-service copybara-service bot deleted the fix-accuracy branch July 9, 2024 17:55
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