diff --git a/docs/_tutorials/pytorch-profiler.md b/docs/_tutorials/pytorch-profiler.md index 6e32d72c2aa8..113061fae4e6 100644 --- a/docs/_tutorials/pytorch-profiler.md +++ b/docs/_tutorials/pytorch-profiler.md @@ -49,7 +49,7 @@ The `record_function` context manager can be used to label arbitrary code ranges ```python with profile(record_shapes=True) as prof: # record_shapes indicates whether to record shapes of the operator inputs. - with record_function("""):" + with record_function("model_forward"): model_engine(inputs) ```