Fix *_batch_end typos (Lightning-AI#17188)

Co-authored-by: Carlos Mocholí <[email protected]>
Git-abouvier · Mar 25, 2023 · 3458258 · 3458258
1 parent 7ec9a97
commit 3458258
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docs/source-pytorch/upgrade/sections/1_9_advanced.rst b/docs/source-pytorch/upgrade/sections/1_9_advanced.rst
@@ -231,15 +231,15 @@
      - `PR16750`_
 
    * - had any logic except reducing the DP outputs in  ``LightningModule.training_step_end`` hook
-     - port it to ``LightningModule.training_batch_end`` hook
+     - port it to ``LightningModule.on_train_batch_end`` hook
      - `PR16791`_
 
    * - had any logic except reducing the DP outputs in  ``LightningModule.validation_step_end`` hook
-     - port it to ``LightningModule.validation_batch_end`` hook
+     - port it to ``LightningModule.on_validation_batch_end`` hook
      - `PR16791`_
 
    * - had any logic except reducing the DP outputs in  ``LightningModule.test_step_end`` hook
-     - port it to ``LightningModule.test_batch_end`` hook
+     - port it to ``LightningModule.on_test_batch_end`` hook
      - `PR16791`_
 
    * - used ``pl.strategies.DDPSpawnStrategy``