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