예제 #1
0
 def get_loggers(self):
     return {
         "console": dl.ConsoleLogger(),
         "csv": dl.CSVLogger(logdir=self._logdir),
         "tensorboard": dl.TensorboardLogger(logdir=self._logdir),
         "mlflow": dl.MLflowLogger(experiment=self._name),
     }
예제 #2
0
    def get_loggers(self):
        loggers = {
            "console": dl.ConsoleLogger(),
            "csv": dl.CSVLogger(logdir=self._logdir),
            "tensorboard": dl.TensorboardLogger(logdir=self._logdir),
        }
        if SETTINGS.mlflow_required:
            loggers["mlflow"] = dl.MLflowLogger(experiment=self.name)

        if SETTINGS.wandb_required:
            loggers["wandb"] = dl.WandbLogger(project="catalyst_test", name=self.name)

        if SETTINGS.neptune_required:
            loggers["neptune"] = dl.NeptuneLogger(
                base_namespace="catalyst-tests",
                api_token="ANONYMOUS",
                project="common/catalyst-integration",
            )

        return loggers