diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py index 9838ae55a5..e9c68fd0ae 100644 --- a/pytorch_lightning/root_module/model_saving.py +++ b/pytorch_lightning/root_module/model_saving.py @@ -1,6 +1,7 @@ import torch import os import re +import pdb class ModelIO(object): @@ -102,6 +103,7 @@ class TrainerIO(object): # save exp to make sure we get all the metrics experiment.save() + pdb.set_trace() ckpt_number = self.max_ckpt_in_folder(folderpath) + 1 if not os.path.exists(folderpath):