Skip to content

Conversation

sadra-barikbin
Copy link
Collaborator

@sadra-barikbin sadra-barikbin commented Dec 8, 2022

  • compute method returns float in PSNR and SSIM as it does in other metrics.
  • Refactor tests in PSNR
  • Refactor tests in SSIM

@github-actions github-actions bot added the module: metrics Metrics module label Dec 8, 2022
@sadra-barikbin sadra-barikbin changed the title Refactor PSNR Refactor PSNR and SSIM Dec 8, 2022
@vfdev-5
Copy link
Collaborator

vfdev-5 commented Dec 8, 2022

@sadra-barikbin can you provide PR description with the info we decided in discord.

@vfdev-5
Copy link
Collaborator

vfdev-5 commented Dec 8, 2022

Let's also swtich SSIM back to float as well

Copy link
Collaborator

@vfdev-5 vfdev-5 left a comment

Choose a reason for hiding this comment

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

Thanks, LGTM!

However we need to fix horovod tests which are skipped right now:

tests/ignite/metrics/test_psnr.py::test_distrib_input_float[horovod] SKIPPED (Skip if no Horovod dist support) [ 86%]

https://github.com/pytorch/ignite/actions/runs/3654919549/jobs/6175755766#step:9:1770

@sadra-barikbin
Copy link
Collaborator Author

Don't you merge it?

@vfdev-5 vfdev-5 merged commit 7c8d801 into pytorch:master Dec 10, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
module: metrics Metrics module
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants