fix visualizer

2021-03-31 19:32:23 +08:00 · 2021-03-31 19:32:23 +08:00 · 9798d07337
parent f84d460613
commit 9798d07337
1 changed files with 1 additions and 1 deletions
--- a/parakeet/training/experiment.py
+++ b/parakeet/training/experiment.py
@ -237,7 +237,7 @@ class ExperimentBase(object):
        unexpected behaviors.
        """
        # visualizer
-        visualizer = SummaryWriter(logdir=str(self.output_dir))
+        visualizer = LogWriter(logdir=str(self.output_dir))

        self.visualizer = visualizer