Unverified Commit 1a66977f authored by ritosonn's avatar ritosonn Committed by GitHub
Browse files

[fix]: Fix a bug where logs are missing when two or more loggers were set (#1015)

* +super().before_run()

* flip reset_flag
parent b8c09f3b
......@@ -13,7 +13,7 @@ class MlflowLoggerHook(LoggerHook):
log_model=True,
interval=10,
ignore_last=True,
reset_flag=True,
reset_flag=False,
by_epoch=True):
"""Class to log metrics and (optionally) a trained model to MLflow.
......@@ -60,6 +60,7 @@ class MlflowLoggerHook(LoggerHook):
@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:
......
......@@ -22,7 +22,7 @@ class PaviLoggerHook(LoggerHook):
add_last_ckpt=False,
interval=10,
ignore_last=True,
reset_flag=True,
reset_flag=False,
by_epoch=True,
img_key='img_info'):
super(PaviLoggerHook, self).__init__(interval, ignore_last, reset_flag,
......@@ -34,6 +34,7 @@ class PaviLoggerHook(LoggerHook):
@master_only
def before_run(self, runner):
super(PaviLoggerHook, self).before_run(runner)
try:
from pavi import SummaryWriter
except ImportError:
......
......@@ -14,7 +14,7 @@ class TensorboardLoggerHook(LoggerHook):
log_dir=None,
interval=10,
ignore_last=True,
reset_flag=True,
reset_flag=False,
by_epoch=True):
super(TensorboardLoggerHook, self).__init__(interval, ignore_last,
reset_flag, by_epoch)
......@@ -22,6 +22,7 @@ class TensorboardLoggerHook(LoggerHook):
@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
......
......@@ -11,7 +11,7 @@ class WandbLoggerHook(LoggerHook):
init_kwargs=None,
interval=10,
ignore_last=True,
reset_flag=True,
reset_flag=False,
commit=True,
by_epoch=True,
with_step=True):
......@@ -32,6 +32,7 @@ class WandbLoggerHook(LoggerHook):
@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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment