lightning/tests/metrics
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
..
classification
functional
regression Add `dim` to `pytorch_lightning.metrics.PSNR` (#5957) 2021-02-17 11:55:40 +01:00
__init__.py
test_composition.py fixing some compatibility with PT 1.8 (#5864) 2021-02-09 18:25:57 +01:00
test_ddp.py
test_metric.py
test_metric_lightning.py
utils.py