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

Reproducibility Issues with SSIM and MS-SSIM as Loss Functions #56

Open
batman47steam opened this issue Jun 9, 2024 · 0 comments
Open

Comments

@batman47steam
Copy link

I would like to ask about the reproducibility issues with SSIM and MS-SSIM. I am considering using MS-SSIM as a loss function to supervise my model's training. However, I have noticed that when using pytorch_msssim for SSIM and MS-SSIM, the training loss fluctuates with each run. In contrast, using the SSIM implementation from this source does not result in such fluctuations. Have you encountered similar issues, or do you have any suggestions to ensure consistent loss during training?

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

No branches or pull requests

1 participant