Back to Pytorch Lightning

Find bottlenecks in your code

docs/source-pytorch/tuning/profiler.rst

2.6.41.0 KB
Original Source

.. _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>