+super().before_run()

open-mmlab · May 9, 2021 · 476adf7 · 476adf7
1 parent db6b054
commit 476adf7
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git a/mmcv/runner/hooks/logger/mlflow.py b/mmcv/runner/hooks/logger/mlflow.py
@@ -60,6 +60,7 @@ def import_mlflow(self):
 
     @master_only
     def before_run(self, runner):
+        super(MlflowLoggerHook, self).before_run(runner)
         if self.exp_name is not None:
             self.mlflow.set_experiment(self.exp_name)
         if self.tags is not None:

diff --git a/mmcv/runner/hooks/logger/pavi.py b/mmcv/runner/hooks/logger/pavi.py
@@ -34,6 +34,7 @@ def __init__(self,
 
     @master_only
     def before_run(self, runner):
+        super(PaviLoggerHook, self).before_run(runner)
         try:
             from pavi import SummaryWriter
         except ImportError:

diff --git a/mmcv/runner/hooks/logger/tensorboard.py b/mmcv/runner/hooks/logger/tensorboard.py
@@ -22,6 +22,7 @@ def __init__(self,
 
     @master_only
     def before_run(self, runner):
+        super(TensorboardLoggerHook, self).before_run(runner)
         if TORCH_VERSION < '1.1' or TORCH_VERSION == 'parrots':
             try:
                 from tensorboardX import SummaryWriter

diff --git a/mmcv/runner/hooks/logger/wandb.py b/mmcv/runner/hooks/logger/wandb.py
@@ -32,6 +32,7 @@ def import_wandb(self):
 
     @master_only
     def before_run(self, runner):
+        super(WandbLoggerHook, self).before_run(runner)
         if self.wandb is None:
             self.import_wandb()
         if self.init_kwargs: