Fix --reset-meters

Summary: Pull Request resolved: https://github.com/fairinternal/fairseq-py/pull/756 Differential Revision: D16418302 Pulled By: myleott fbshipit-source-id: 62495a0bff41d1741e2b09807a3b43ff2c66c8fb

Fix --reset-meters
Summary: Pull Request resolved: https://github.com/fairinternal/fairseq-py/pull/756 Differential Revision: D16418302 Pulled By: myleott fbshipit-source-id: 62495a0bff41d1741e2b09807a3b43ff2c66c8fb
906411da · Myle Ott · Facebook Github Bot · bccfa7d0 · 906411da
Commit 906411da authored Jul 22, 2019 by Myle Ott Committed by Facebook Github Bot Jul 22, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

fairseq/trainer.py fairseq/trainer.py +1 -1

No files found.
--- a/fairseq/trainer.py
+++ b/fairseq/trainer.py
@@ -184,7 +184,7 @@ class Trainer(object):
            self.lr_step(epoch)
-            if 'train_meters' in extra_state:
+            if 'train_meters' in extra_state and not reset_meters:
                self.meters.update(extra_state['train_meters'])
                del extra_state['train_meters']