Add logger property (#1372)

* added logger property for all loggers, except mlflow

* added logger property for mlflow

* added NotImplementedError in case if internal logger is unavailable

* updated CHANGELOG.md

Author: y-ksenia

CHANGELOG.md

@@ -9,6 +9,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 ### Added
 
 - MNIST dataset for SSL banchmark ([#1368](https://github.com/catalyst-team/catalyst/pull/1368))
+- logger property for logging customization ([#1372](https://github.com/catalyst-team/catalyst/pull/1372))
 
 ### Changed
 

catalyst/core/logger.py

@@ -1,4 +1,4 @@
-from typing import Dict
+from typing import Any, Dict
 
 import numpy as np
 
@@ -22,6 +22,15 @@ def __init__(self, log_batch_metrics: bool, log_epoch_metrics: bool) -> None:
         self._log_batch_metrics = log_batch_metrics
         self._log_epoch_metrics = log_epoch_metrics
 
+    @property
+    def logger(self) -> Any:
+        """Internal logger/experiment/etc. from the monitoring system. # noqa: DAR401
+
+        Returns: # noqa: DAR201, DAR202
+            Any: internal logger/experiment/etc. from the monitoring system.
+        """
+        raise NotImplementedError
+
     @property
     def log_batch_metrics(self) -> bool:
         """Boolean flag to log batch metrics.

catalyst/loggers/comet.py

@@ -117,6 +117,11 @@ def __init__(
         if tags is not None:
             self.experiment.add_tags(tags)
 
+    @property
+    def logger(self):
+        """Internal logger/experiment/etc. from the monitoring system."""
+        return self.experiment
+
     def _get_experiment(self, mode, experiment_id=None):
         if mode == "offline":
             if experiment_id is not None:

catalyst/loggers/csv.py

@@ -79,6 +79,11 @@ def __init__(self, logdir: str, use_logdir_postfix: bool = False):
         self.loggers = {}
         os.makedirs(self.logdir, exist_ok=True)
 
+    @property
+    def logger(self):
+        """Internal logger/experiment/etc. from the monitoring system."""
+        return self.loggers
+
     def _make_header(self, metrics: Dict[str, float], loader_key: str):
         log_line_header = "step,"
         for metric in sorted(metrics.keys()):

catalyst/loggers/mlflow.py

@@ -159,6 +159,11 @@ def __init__(
         mlflow.set_experiment(self.experiment)
         _get_or_start_run(run_name=self.run)
 
+    @property
+    def logger(self):
+        """Internal logger/experiment/etc. from the monitoring system."""
+        return mlflow
+
     @staticmethod
     def _log_metrics(metrics: Dict[str, float], step: int, loader_key: str, suffix=""):
         for key, value in metrics.items():

catalyst/loggers/neptune.py

@@ -152,6 +152,11 @@ def __init__(
         except (ImportError, NameError, AttributeError):
             pass
 
+    @property
+    def logger(self):
+        """Internal logger/experiment/etc. from the monitoring system."""
+        return self.run
+
     def _log_metrics(self, metrics: Dict[str, float], neptune_path: str, step: int):
         for key, value in metrics.items():
             self.run[f"{neptune_path}/{key}"].log(value=float(value), step=step)

catalyst/loggers/tensorboard.py

@@ -94,6 +94,11 @@ def __init__(
         self.loggers = {}
         os.makedirs(self.logdir, exist_ok=True)
 
+    @property
+    def logger(self):
+        """Internal logger/experiment/etc. from the monitoring system."""
+        return self.loggers
+
     def _check_loader_key(self, loader_key: str):
         if loader_key not in self.loggers.keys():
             logdir = os.path.join(self.logdir, f"{loader_key}")

catalyst/loggers/wandb.py

@@ -94,6 +94,11 @@ def __init__(
             project=self.project, name=self.name, entity=self.entity, allow_val_change=True
         )
 
+    @property
+    def logger(self):
+        """Internal logger/experiment/etc. from the monitoring system."""
+        return self.run
+
     def _log_metrics(self, metrics: Dict[str, float], step: int, loader_key: str, prefix=""):
         for key, value in metrics.items():
             self.run.log({f"{key}_{prefix}/{loader_key}": value}, step=step)