Commit 83053f97 authored by Myle Ott's avatar Myle Ott
Browse files

Fixes for `--log-format`

parent 55a989e8
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
@@ -80,7 +80,7 @@ class json_progress_bar(progress_bar):
    def __init__(self, iterable, epoch=None, prefix=None, log_interval=1000):
        super().__init__(iterable, epoch, prefix)
        self.log_interval = log_interval
        self.postfix_json = None
        self.stats = None

    def __iter__(self):
        size = float(len(self.iterable))
+1 −1
Changes for train.py: 1 added line, 1 removed line.
Original line number Diff line number Diff line
@@ -36,7 +36,7 @@ def main():

    args = utils.parse_args_and_arch(parser)

    if args.no_progress_bar:
    if args.no_progress_bar and args.log_format == 'tqdm':
        args.log_format = 'simple'

    if not os.path.exists(args.save_dir):