File tree Expand file tree Collapse file tree 1 file changed +18
-18
lines changed Expand file tree Collapse file tree 1 file changed +18
-18
lines changed Original file line number Diff line number Diff line change 5
5
"""
6
6
7
7
######################################################################
8
- """
9
- This recipe explains how to use PyTorch profiler and measure the time and
10
- memory consumption of the model's operators.
8
+ #
9
+ # This recipe explains how to use PyTorch profiler and measure the time and
10
+ # memory consumption of the model's operators.
11
11
12
- Introduction
13
- ------------
14
- PyTorch includes a simple profiler API that is useful when user needs
15
- to determine the most expensive operators in the model.
12
+ # Introduction
13
+ # ------------
14
+ # PyTorch includes a simple profiler API that is useful when user needs
15
+ # to determine the most expensive operators in the model.
16
16
17
- In this recipe, we will use a simple Resnet model to demonstrate how to
18
- use profiler to analyze model performance.
17
+ # In this recipe, we will use a simple Resnet model to demonstrate how to
18
+ # use profiler to analyze model performance.
19
19
20
- Prerequisites
21
- ---------------
22
- - ``torch >= 1.9``
20
+ # Prerequisites
21
+ # ---------------
22
+ # - ``torch >= 1.9``
23
23
24
- Setup
25
- -----
26
- To install ``torch`` and ``torchvision`` use the following command:
24
+ # Setup
25
+ # -----
26
+ # To install ``torch`` and ``torchvision`` use the following command:
27
27
28
- .. code-block:: sh
28
+ # .. code-block:: sh
29
29
30
- pip install torch torchvision
30
+ # pip install torch torchvision
31
31
32
- """
32
+ #
33
33
34
34
######################################################################
35
35
# Steps
You can’t perform that action at this time.
0 commit comments