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

View File

@ -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):