lightning/tests/base
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 [tests/core] Updated with BoringModel and added BoringDataModule (#5432) 2021-01-13 01:48:37 -05:00
boring_model.py [tests/core] Updated with BoringModel and added BoringDataModule (#5432) 2021-01-13 01:48:37 -05:00
dataloaders.py notices (#4118) 2020-10-13 07:18:07 -04:00
datamodules.py Fix pre-commit isort failure on tests/base/*.py (#5429) 2021-01-15 03:01:57 -05:00
datasets.py Fix pre-commit isort failure on tests/base/*.py (#5429) 2021-01-15 03:01:57 -05:00
deterministic_model.py Fix pre-commit isort failure on tests/base/*.py (#5429) 2021-01-15 03:01:57 -05:00
develop_pipelines.py Refactor: clean trainer device & distrib getters (#5300) 2021-01-12 05:22:37 -05:00
develop_utils.py [tests/core] Updated with BoringModel and added BoringDataModule (#5432) 2021-01-13 01:48:37 -05:00
model_optimizers.py Refactor/prune unused EvalModel methods (#5331) 2021-01-03 15:09:54 +01:00
model_template.py Fix pre-commit isort failure on tests/base/*.py (#5429) 2021-01-15 03:01:57 -05:00
model_test_dataloaders.py Fix pre-commit isort failure on tests/base/*.py (#5429) 2021-01-15 03:01:57 -05:00
model_test_epoch_ends.py Refactor: clean trainer device & distrib getters (#5300) 2021-01-12 05:22:37 -05:00
model_test_steps.py [feat] Add PyTorch Profiler. (#5560) 2021-01-26 06:48:54 -05:00
model_train_dataloaders.py Fix pre-commit isort failure on tests/base/*.py (#5429) 2021-01-15 03:01:57 -05:00
model_train_steps.py prune deprecated EvalResult (#5633) 2021-01-26 03:09:39 -05:00
model_utilities.py notices (#4118) 2020-10-13 07:18:07 -04:00
model_valid_dataloaders.py Fix pre-commit isort failure on tests/base/*.py (#5429) 2021-01-15 03:01:57 -05:00
model_valid_epoch_ends.py Refactor/prune unused EvalModel methods (#5331) 2021-01-03 15:09:54 +01:00
model_valid_steps.py prune deprecated EvalResult (#5633) 2021-01-26 03:09:39 -05:00
models.py Fix pre-commit isort failure on tests/base/*.py (#5429) 2021-01-15 03:01:57 -05:00
simple_model.py Fix pre-commit isort failure on tests/base/*.py (#5429) 2021-01-15 03:01:57 -05:00
test_datasets.py Fix pre-commit isort failure on tests/base/*.py (#5429) 2021-01-15 03:01:57 -05:00