fixed epoch continuation from checkpoint

2019-05-05 12:15:04 -04:00 · 2019-05-05 12:15:04 -04:00 · 12352f1949
parent f881bf6750
commit 12352f1949
1 changed files with 1 additions and 0 deletions
--- a/pytorch_lightning/root_module/model_saving.py
+++ b/pytorch_lightning/root_module/model_saving.py
@ -88,6 +88,7 @@ class TrainerIO(object):
        self.early_stop_callback.wait = checkpoint['early_stop_callback_wait']
        self.early_stop_callback.patience = checkpoint['early_stop_callback_patience']
        self.global_step = checkpoint['global_step']
+        self.current_epoch = checkpoint['epoch']

        # restore the optimizers
        optimizer_states = checkpoint['optimizer_states']