Skip to content

GitLab

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

Fixes for `--log-format`

parent 55a989e8
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
fairseq/progress_bar.py
View file @ 83053f97
...@@ -80,7 +80,7 @@ class json_progress_bar(progress_bar): ...@@ -80,7 +80,7 @@ class json_progress_bar(progress_bar):
def __init__(self, iterable, epoch=None, prefix=None, log_interval=1000): def __init__(self, iterable, epoch=None, prefix=None, log_interval=1000):
super().__init__(iterable, epoch, prefix) super().__init__(iterable, epoch, prefix)
self.log_interval = log_interval self.log_interval = log_interval
self.postfix_json = None self.stats = None
def __iter__(self): def __iter__(self):
size = float(len(self.iterable)) size = float(len(self.iterable))
......
train.py
View file @ 83053f97
...@@ -36,7 +36,7 @@ def main(): ...@@ -36,7 +36,7 @@ def main():
args = utils.parse_args_and_arch(parser) 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' args.log_format = 'simple'
if not os.path.exists(args.save_dir): if not os.path.exists(args.save_dir):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment