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

Fix Constant Pruning Modifier #1751

Merged
merged 1 commit into from
Oct 6, 2023
Merged

Fix Constant Pruning Modifier #1751

merged 1 commit into from
Oct 6, 2023

Conversation

Satrat
Copy link

@Satrat Satrat commented Oct 6, 2023

The mask calculation in the constant pruning modifier was inverted, this one character PR fixes it!

Testing

Constant pruning modifier is tested in this branch: https://github.com/neuralmagic/sparseml/tree/modifier-refactor-constant-pruning-bug

Run python bug_constant_pruning to confirm constant pruning modifier is working as intended

@rahul-tuli
Copy link
Member

Let's Go!!

@Satrat Satrat merged commit 528289b into main Oct 6, 2023
11 checks passed
@Satrat Satrat deleted the fix_inverted_mask branch October 6, 2023 17:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants