Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
lightning/pytorch_lightning/profiler/__init__.py

16 lines
514 B
Python
Raw Normal View History

[doc] Move each profiler to its own file + Add missing PyTorchProfiler to the doc (#7822)
2021-06-04 15:38:29 +00:00
from pytorch_lightning.profiler.advanced import AdvancedProfiler
from pytorch_lightning.profiler.base import AbstractProfiler, BaseProfiler, PassThroughProfiler
split profilers (#6261)
2021-03-02 07:57:49 +00:00
from pytorch_lightning.profiler.pytorch import PyTorchProfiler
[doc] Move each profiler to its own file + Add missing PyTorchProfiler to the doc (#7822)
2021-06-04 15:38:29 +00:00
from pytorch_lightning.profiler.simple import SimpleProfiler
XLA Profiler integration (#8014)
2021-06-28 19:28:05 +00:00
from pytorch_lightning.profiler.xla import XLAProfiler
new feature for profiling training runs (#782) * initial implementation * formatting, pass through profiler, docstring * call profiler during training * add initial tests * report stats when training is done * fix formatting * error handling, bugfix in passthroughprofiler * finish documenting profiler arg in Trainer * relax required precision for profiling tests * option to dump cProfiler results to text file * use logging, format with black * include profiler in docs * improved logging and better docs * appease the linter * better summaries, wrapper for iterables * fix typo * allow profiler=True creation * more documentation * add tests for advanced profiler * Update trainer.py * make profilers accessible in pl.utilities * reorg profiler files * change import for profiler tests Co-authored-by: William Falcon <waf2107@columbia.edu>
2020-02-07 03:01:21 +00:00
__all__ = [
Replace `yapf` with `black` (#7783) Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2021-07-26 11:37:35 +00:00
"AbstractProfiler",
"BaseProfiler",
"AdvancedProfiler",
"PassThroughProfiler",
"PyTorchProfiler",
"SimpleProfiler",
"XLAProfiler",
new feature for profiling training runs (#782) * initial implementation * formatting, pass through profiler, docstring * call profiler during training * add initial tests * report stats when training is done * fix formatting * error handling, bugfix in passthroughprofiler * finish documenting profiler arg in Trainer * relax required precision for profiling tests * option to dump cProfiler results to text file * use logging, format with black * include profiler in docs * improved logging and better docs * appease the linter * better summaries, wrapper for iterables * fix typo * allow profiler=True creation * more documentation * add tests for advanced profiler * Update trainer.py * make profilers accessible in pl.utilities * reorg profiler files * change import for profiler tests Co-authored-by: William Falcon <waf2107@columbia.edu>
2020-02-07 03:01:21 +00:00
]