Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

adding support for interrupt signals

This commit is contained in:
William Falcon 2019-06-14 09:44:19 -04:00
parent 32edc6d7b7
commit edf03063a1
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/root_module/model_saving.py
View File

@ -1,6 +1,7 @@
import torch import torch
import os import os
import re import re
import pdb
class ModelIO(object): class ModelIO(object):
@ -102,6 +103,7 @@ class TrainerIO(object):
# save exp to make sure we get all the metrics # save exp to make sure we get all the metrics
experiment.save() experiment.save()
pdb.set_trace()
ckpt_number = self.max_ckpt_in_folder(folderpath) + 1 ckpt_number = self.max_ckpt_in_folder(folderpath) + 1
if not os.path.exists(folderpath): if not os.path.exists(folderpath):