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

Update Quantization Logging to New Framework #2313

Merged
merged 8 commits into from
Jun 10, 2024
Merged

Conversation

Satrat
Copy link

@Satrat Satrat commented May 30, 2024

The quantization percentage logging has been reporting 0.0% quantization since we switched over to the new framework. This PR updates the calculation to use the new framework. It also cleans up the string formatting and adds Embedding and a prunable/quantizable layer so the percentages are more accurate

Before

2024-05-30 16:04:21 sparseml.transformers.finetune.session_mixin INFO     There are 1034420224 prunable params which have 1.2609391906088643 avg sparsity.
2024-05-30 16:04:21 sparseml.transformers.finetune.session_mixin INFO     There are 1034420224 quantizable params, with a quantization percentage of 0.0.

After

2024-05-30 15:57:12 sparseml.transformers.finetune.session_mixin INFO     There are 1099956224 prunable params which have 1.19% avg sparsity.
2024-05-30 15:57:12 sparseml.transformers.finetune.session_mixin INFO     There are 1099956224 quantizable params, with a quantization percentage of 88.08%.

bfineran
bfineran previously approved these changes May 30, 2024
rahul-tuli
rahul-tuli previously approved these changes May 30, 2024
@Satrat Satrat dismissed stale reviews from rahul-tuli and bfineran via d768716 May 30, 2024 18:27
Copy link
Contributor

@dbogunowicz dbogunowicz left a comment

Choose a reason for hiding this comment

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

awesome!

@Satrat Satrat merged commit 934f0d8 into main Jun 10, 2024
17 checks passed
@Satrat Satrat deleted the sa/fix_quantize_logging branch June 10, 2024 15:39
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.

4 participants