From c9a1629900f4f381f78dcfb8f7ad26f8844c63aa Mon Sep 17 00:00:00 2001 From: David de la Iglesia Castro Date: Tue, 26 Jul 2022 17:21:35 +0200 Subject: [PATCH] Update mmcv/runner/hooks/logger/dvclive.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> --- mmcv/runner/hooks/logger/dvclive.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/mmcv/runner/hooks/logger/dvclive.py b/mmcv/runner/hooks/logger/dvclive.py index cff88b823b..7a3e296f54 100644 --- a/mmcv/runner/hooks/logger/dvclive.py +++ b/mmcv/runner/hooks/logger/dvclive.py @@ -57,9 +57,10 @@ def import_dvclive(self, **kwargs) -> None: def log(self, runner) -> None: tags = self.get_loggable_tags(runner) if tags: - self.dvclive.set_step( - self.get_epoch(runner) if self.by_epoch else self. - get_iter(runner)) + step = self.get_epoch(runner) if self.by_epoch else self.get_iter( + runner) + self.dvclive.set_step(step) + for k, v in tags.items(): self.dvclive.log(k, v)