Fix docstring (Lightning-AI#2884)

* Fix docstring "mean absolute loss" rather than "root mean absolute loss" * minor docstring fix Co-authored-by: rohitgr7 <rohitgr1998@gmail.com>
f4hy · Aug 8, 2020 · 6ebe0d7 · 6ebe0d7
1 parent f798cff
commit 6ebe0d7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pytorch_lightning/callbacks/base.py b/pytorch_lightning/callbacks/base.py
@@ -47,11 +47,11 @@ def on_sanity_check_end(self, trainer, pl_module):
         pass
 
     def on_train_batch_start(self, trainer, pl_module, batch, batch_idx, dataloader_idx):
-        """Called when the validation batch begins."""
+        """Called when the train batch begins."""
         pass
 
     def on_train_batch_end(self, trainer, pl_module, batch, batch_idx, dataloader_idx):
-        """Called when the validation batch ends."""
+        """Called when the train batch ends."""
         pass
 
     def on_train_epoch_start(self, trainer, pl_module):

diff --git a/pytorch_lightning/metrics/regression.py b/pytorch_lightning/metrics/regression.py
@@ -101,7 +101,7 @@ def forward(self, pred: torch.Tensor, target: torch.Tensor) -> torch.Tensor:
 
 class MAE(Metric):
     """
-    Computes the root mean absolute loss or L1-loss.
+    Computes the mean absolute loss or L1-loss.
 
     Example: