Old checkpoints can't be loaded because of a new meter

c9b800d2 · Sergey Edunov · Myle Ott · 9c102784 · c9b800d2
Commit c9b800d2 authored Aug 24, 2018 by Sergey Edunov Committed by Myle Ott Sep 03, 2018
Hide 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
@@ -107,7 +107,7 @@ class Trainer(object):
            self._num_updates = last_optim['num_updates']

        if extra_state is not None and 'train_meters' in extra_state:
-            self.meters = extra_state['train_meters']
+            self.meters.update(extra_state['train_meters'])
            del extra_state['train_meters']

            # reset TimeMeters, since their start times don't make sense anymore