lightning/tests/core
chaton 5f3372871a
[feat] Add PyTorch Profiler. (#5560)
* add profiler

* add profiler

* update

* resolve flake8

* update doc

* update changelog

* clean doc

* delete prof file

* merge pr codebase

* update

* update doc

* update doc

* update doc

* update on comments

* update docstring

* update docstring

* try

* update test

* Update pytorch_lightning/profiler/__init__.py

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

* Update pytorch_lightning/profiler/__init__.py

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

* update on comments

* remove old code

* add support for ddp

* resolve flake8

* Update pytorch_lightning/profiler/__init__.py

Co-authored-by: Sean Naren <sean.narenthiran@gmail.com>

* resolve tests

* resolve flake8

Co-authored-by: Jirka Borovec <jirka.borovec@seznam.cz>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Sean Naren <sean.narenthiran@gmail.com>
2021-01-26 06:48:54 -05:00
..
__init__.py Refactor model summary + generalize example input array (#1773) 2020-06-15 17:05:58 -04:00
test_datamodules.py [tests/core] Updated with BoringModel and added BoringDataModule (#5432) 2021-01-13 01:48:37 -05:00
test_decorators.py [tests/core] Updated with BoringModel and added BoringDataModule (#5432) 2021-01-13 01:48:37 -05:00
test_lightning_module.py [tests/core] Updated with BoringModel and added BoringDataModule (#5432) 2021-01-13 01:48:37 -05:00
test_lightning_optimizer.py [tests/core] Updated with BoringModel and added BoringDataModule (#5432) 2021-01-13 01:48:37 -05:00
test_memory.py [feat] Add PyTorch Profiler. (#5560) 2021-01-26 06:48:54 -05:00
test_metric_result_integration.py [tests/core] Updated with BoringModel and added BoringDataModule (#5432) 2021-01-13 01:48:37 -05:00
test_results.py prune deprecated EvalResult (#5633) 2021-01-26 03:09:39 -05:00