mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-09 11:32:07 +08:00
adding support for interrupt signals
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user