feat: multi-scale SSIM#361
Merged
johannaSommer merged 4 commits intoPrunaAI:mainfrom Oct 10, 2025
R055A:feat/multi-scale-ssim
Merged
feat: multi-scale SSIM#361johannaSommer merged 4 commits intoPrunaAI:mainfrom R055A:feat/multi-scale-ssim
johannaSommer merged 4 commits intoPrunaAI:mainfrom
R055A:feat/multi-scale-ssim
Conversation
begumcig
requested changes
Oct 1, 2025
Contributor
Author
johannaSommer
approved these changes
Oct 7, 2025
begumcig
approved these changes
Oct 10, 2025
Member
begumcig
left a comment
There was a problem hiding this comment.
Thank you so much for the updates, looks great!
Contributor
Author
Thank you. I just now fixed the lint error |
Member
|
Test failures are due to the HF servers and not related to this PR, ready to merge ☑️ |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Description
Add feature: multi-scale SSIM
Related Issue
Fixes #359
Type of Change
How Has This Been Tested?
Unit testing
Checklist
Additional Notes
Note
Adds
msssimmetric viaTorchMetricWrapper, generalizesssim_updateto support SSIM and MS-SSIM, and adds targeted tests (type mapping, invalid params).MultiScaleStructuralSimilarityIndexMeasureimport and new enum entryTorchMetrics.msssimusingpairwise_y_gtand sharedssim_update.ssim_updateto acceptStructuralSimilarityIndexMeasure | MultiScaleStructuralSimilarityIndexMeasureand handle float casting.TorchMetricWrapper('ssim')and('msssim')instantiate the correct metric types.ValueError.Written by Cursor Bugbot for commit 5e8558f. This will update automatically on new commits. Configure here.