Trainers: fix support for non-TensorBoardLogger (#1145)

* Trainers: fix support for CSVLogger * No need to check truthiness of attr * Revert "No need to check truthiness of attr" This reverts commit 450df6c.
microsoft · Apr 10, 2023 · 6b1c2cc · 6b1c2cc
1 parent 0a266b2
commit 6b1c2cc
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 0 deletions.
diff --git a/torchgeo/trainers/classification.py b/torchgeo/trainers/classification.py
@@ -192,6 +192,7 @@ def validation_step(self, *args: Any, **kwargs: Any) -> None:
             and hasattr(self.trainer, "datamodule")
             and self.logger
             and hasattr(self.logger, "experiment")
+            and hasattr(self.logger.experiment, "add_figure")
         ):
             try:
                 datamodule = self.trainer.datamodule
@@ -376,6 +377,7 @@ def validation_step(self, *args: Any, **kwargs: Any) -> None:
             and hasattr(self.trainer, "datamodule")
             and self.logger
             and hasattr(self.logger, "experiment")
+            and hasattr(self.logger.experiment, "add_figure")
         ):
             try:
                 datamodule = self.trainer.datamodule

diff --git a/torchgeo/trainers/detection.py b/torchgeo/trainers/detection.py
@@ -230,6 +230,7 @@ def validation_step(self, *args: Any, **kwargs: Any) -> None:
             and hasattr(self.trainer, "datamodule")
             and self.logger
             and hasattr(self.logger, "experiment")
+            and hasattr(self.logger.experiment, "add_figure")
         ):
             try:
                 datamodule = self.trainer.datamodule

diff --git a/torchgeo/trainers/regression.py b/torchgeo/trainers/regression.py
@@ -149,6 +149,7 @@ def validation_step(self, *args: Any, **kwargs: Any) -> None:
             and hasattr(self.trainer, "datamodule")
             and self.logger
             and hasattr(self.logger, "experiment")
+            and hasattr(self.logger.experiment, "add_figure")
         ):
             try:
                 datamodule = self.trainer.datamodule

diff --git a/torchgeo/trainers/segmentation.py b/torchgeo/trainers/segmentation.py
@@ -202,6 +202,7 @@ def validation_step(self, *args: Any, **kwargs: Any) -> None:
             and hasattr(self.trainer, "datamodule")
             and self.logger
             and hasattr(self.logger, "experiment")
+            and hasattr(self.logger.experiment, "add_figure")
         ):
             try:
                 datamodule = self.trainer.datamodule