lightning/docs/source-pytorch/tuning/profiler.rst

50 lines
1.0 KiB
ReStructuredText

.. _profiler:
#############################
Find bottlenecks in your code
#############################
.. raw:: html
<div class="display-card-container">
<div class="row">
.. Add callout items below this line
.. displayitem::
:header: Basic
:description: Learn to find bottlenecks in the training loop.
:col_css: col-md-3
:button_link: profiler_basic.html
:height: 150
:tag: basic
.. displayitem::
:header: Intermediate
:description: Learn to find bottlenecks in PyTorch operations.
:col_css: col-md-3
:button_link: profiler_intermediate.html
:height: 150
:tag: intermediate
.. displayitem::
:header: Advanced
:description: Learn to profile TPU code.
:col_css: col-md-3
:button_link: profiler_advanced.html
:height: 150
:tag: advanced
.. displayitem::
:header: Expert
:description: Learn to build your own profiler or profile custom pieces of code
:col_css: col-md-3
:button_link: profiler_expert.html
:height: 150
:tag: expert
.. raw:: html
</div>
</div>