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