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", "update",
"rehearse", "rehearse",
"get_loss", "get_loss",
"get_student_teacher_loss", "get_teacher_student_loss",
"initialize", "initialize",
"begin_update", "begin_update",
"finish_update", "finish_update",

View File

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