Commit 0e8414f9 authored by Myle Ott's avatar Myle Ott
Browse files

Remove sweep_log prefix from json progress bar

parent d62a8651
Loading
Loading
Loading
Loading
+1 −1
Changes for fairseq/progress_bar.py: 1 added line, 1 removed line.
Original line number Diff line number Diff line
@@ -118,7 +118,7 @@ class json_progress_bar(progress_bar):
    def print(self, stats):
        """Print end-of-epoch stats."""
        stats = self._format_stats(self.stats, epoch=self.epoch)
        print("sweep_log: " + json.dumps(stats), flush=True)
        print(json.dumps(stats), flush=True)

    def _format_stats(self, stats, epoch=None, update=None):
        postfix = OrderedDict()