Skip to content

Commit ae2ba11

Browse files
committed
Fix typo in pytorch-profiler.md documentation
Corrected the record_function parameter in code example from incorrect '"""):' to 'model_forward'
1 parent 433e3c7 commit ae2ba11

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/_tutorials/pytorch-profiler.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ The `record_function` context manager can be used to label arbitrary code ranges
4949

5050
```python
5151
with profile(record_shapes=True) as prof: # record_shapes indicates whether to record shapes of the operator inputs.
52-
with record_function("""):"
52+
with record_function("model_forward"):
5353
model_engine(inputs)
5454
```
5555

0 commit comments

Comments
 (0)