adding support for interrupt signals
This commit is contained in:
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):
|
||||
|
|
Loading…
Reference in New Issue