Write TimeMeters and StopWatchMeters to tensorboard. (facebookresearc…

…h#34)
taylanbil · Apr 3, 2020 · c252f6f · c252f6f
1 parent aa9d12c
commit c252f6f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/fairseq/progress_bar.py b/fairseq/progress_bar.py
@@ -340,6 +340,10 @@ def _log_to_tensorboard(self, stats, tag='', step=None):
                 if isinstance(logval, torch.Tensor):
                     logval = logval.item()
                 writer.add_scalar(key, logval, step)
+            elif isinstance(stats[key], (TimeMeter, StopwatchMeter)):
+                logval = stats[key].avg
+                if logval:
+                    writer.add_scalar(key, logval, step)
             elif isinstance(stats[key], Number):
                 writer.add_scalar(key, stats[key], step)