From 8d725d9d9ccd3f5bf32aca30c9326049235b59e4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Danie=CC=88l=20de=20Kok?= Date: Fri, 13 Jan 2023 14:37:41 +0100 Subject: [PATCH] Fix incorrect identifier --- spacy/ml/callbacks.py | 2 +- website/docs/api/top-level.mdx | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/spacy/ml/callbacks.py b/spacy/ml/callbacks.py index 9b24c71bd..393f208a6 100644 --- a/spacy/ml/callbacks.py +++ b/spacy/ml/callbacks.py @@ -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", diff --git a/website/docs/api/top-level.mdx b/website/docs/api/top-level.mdx index 2eb3a84af..7e47d324a 100644 --- a/website/docs/api/top-level.mdx +++ b/website/docs/api/top-level.mdx @@ -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 |