diff --git a/docs/source-pytorch/benchmarking/benchmarks.rst b/docs/source-pytorch/benchmarking/benchmarks.rst index af9715fbf4..3ceb187c22 100644 --- a/docs/source-pytorch/benchmarking/benchmarks.rst +++ b/docs/source-pytorch/benchmarking/benchmarks.rst @@ -1,7 +1,7 @@ :orphan: -Benchmark with vanilla PyTorch -============================== +Benchmark performance vs. vanilla PyTorch +========================================= In this section we set grounds for comparison between vanilla PyTorch and PT Lightning for most common scenarios. @@ -17,3 +17,29 @@ In average for simple MNIST CNN classifier we are only about 0.06s slower per ep Learn more about reproducible benchmarking from the `PyTorch Reproducibility Guide `__. + + +---- + +Find performance bottlenecks +============================= + +.. raw:: html + +
+
+ +.. Add callout items below this line + +.. displayitem:: + :header: Find bottlenecks in your models + :description: Benchmark your own Lightning models + :button_link: ../tuning/profiler.html + :col_css: col-md-3 + :height: 180 + :tag: basic + +.. raw:: html + +
+