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