Fix function typo

microsoft · Mar 16, 2023 · b5e5ad6 · b5e5ad6
1 parent cef9bed
commit b5e5ad6
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/torchgeo/trainers/classification.py b/torchgeo/trainers/classification.py
@@ -159,7 +159,7 @@ def training_step(self, *args: Any, **kwargs: Any) -> Tensor:
 
         return cast(Tensor, loss)
 
-    def on_training_epoch_end(self) -> None:
+    def on_train_epoch_end(self) -> None:
         """Logs epoch-level training metrics."""
         self.log_dict(self.train_metrics.compute())
         self.train_metrics.reset()

diff --git a/torchgeo/trainers/regression.py b/torchgeo/trainers/regression.py
@@ -118,7 +118,7 @@ def training_step(self, *args: Any, **kwargs: Any) -> Tensor:
 
         return loss
 
-    def on_training_epoch_end(self) -> None:
+    def on_train_epoch_end(self) -> None:
         """Logs epoch-level training metrics."""
         self.log_dict(self.train_metrics.compute())
         self.train_metrics.reset()

diff --git a/torchgeo/trainers/segmentation.py b/torchgeo/trainers/segmentation.py
@@ -169,7 +169,7 @@ def training_step(self, *args: Any, **kwargs: Any) -> Tensor:
 
         return cast(Tensor, loss)
 
-    def on_training_epoch_end(self) -> None:
+    def on_train_epoch_end(self) -> None:
         """Logs epoch level training metrics."""
         self.log_dict(self.train_metrics.compute())
         self.train_metrics.reset()