Fix incorrect identifier

This commit is contained in:
Daniël de Kok 2023-01-13 14:37:41 +01:00
parent 2774667f77
commit 8d725d9d9c
2 changed files with 2 additions and 2 deletions

View File

@ -23,7 +23,7 @@ DEFAULT_NVTX_ANNOTATABLE_PIPE_METHODS = [
"update",
"rehearse",
"get_loss",
"get_student_teacher_loss",
"get_teacher_student_loss",
"initialize",
"begin_update",
"finish_update",

View File

@ -921,7 +921,7 @@ backprop passes.
Recursively wrap both the models and methods of each pipe using
[NVTX](https://nvidia.github.io/NVTX/) range markers. By default, the following
methods are wrapped: `pipe`, `predict`, `set_annotations`, `update`, `rehearse`,
`get_loss`, `get_student_teacher_loss`, `initialize`, `begin_update`,
`get_loss`, `get_teacher_student_loss`, `initialize`, `begin_update`,
`finish_update`, `update`.
| Name | Description |