training_end to training_step_end

pytorch_lightning/core/lightning.py

@@ -232,7 +232,7 @@ def training_step_end(self, *args, **kwargs):
         on only part of the batch. However, this is still optional
         and only needed for things like softmax or NCE loss.
 
-        If you later switch to ddp or some other mode, this will still be called
+        .. note:: If you later switch to ddp or some other mode, this will still be called
         so that you don't have to change your code
 
         .. code-block:: python
@@ -364,7 +364,7 @@ def validation_step_end(self, *args, **kwargs):
         on only part of the batch. However, this is still optional
         and only needed for things like softmax or NCE loss.
 
-        If you later switch to ddp or some other mode, this will still be called
+        .. note:: If you later switch to ddp or some other mode, this will still be called
         so that you don't have to change your code
 
         .. code-block:: python
@@ -573,7 +573,7 @@ def test_step_end(self, *args, **kwargs):
         on only part of the batch. However, this is still optional
         and only needed for things like softmax or NCE loss.
 
-        If you later switch to ddp or some other mode, this will still be called
+        .. note:: If you later switch to ddp or some other mode, this will still be called
         so that you don't have to change your code
 
         .. code-block:: python