lightning/tests/metrics/regression
manipopopo 6a9cec41a3
Add `dim` to `pytorch_lightning.metrics.PSNR` (#5957)
* Add dim to PSNR

* Update CHANGELOG.md

* Update CHANGELOG.md

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

* Add reduction tests

* Recover warnings on reduction and add tests

* Add copyright texts

* Refactor PSNR

* Change warnings

* Update pytorch_lightning/metrics/functional/psnr.py

Change functional.psnr dim doc

Co-authored-by: Nicki Skafte <skaftenicki@gmail.com>

* Change PSNR dim docs

* Apply suggestions from code review

* tests

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Nicki Skafte <skaftenicki@gmail.com>
Co-authored-by: Jirka Borovec <jirka.borovec@seznam.cz>
2021-02-17 11:55:40 +01:00
..
__init__.py reverted "temporary drop metrics tests while speeding them up" and SKIP (#4115) 2020-10-14 19:01:43 +02:00
test_explained_variance.py yapf tests metrics (#5845) 2021-02-06 11:41:40 -05:00
test_mean_error.py yapf tests metrics (#5845) 2021-02-06 11:41:40 -05:00
test_psnr.py Add `dim` to `pytorch_lightning.metrics.PSNR` (#5957) 2021-02-17 11:55:40 +01:00
test_r2score.py yapf tests metrics (#5845) 2021-02-06 11:41:40 -05:00
test_ssim.py yapf tests metrics (#5845) 2021-02-06 11:41:40 -05:00