mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-26 13:00:58 +08:00
added clean slurm save load test
This commit is contained in:
1 parent
f1de62671d
commit
f5a01edfb8
1 file changed
-1
@@ -176,7 +176,6 @@ class TrainerIO(object):
|
||||
model = self.__get_model()
|
||||
|
||||
# load the state_dict on the model automatically
|
||||
pdb.set_trace()
|
||||
model.load_state_dict(checkpoint['state_dict'])
|
||||
|
||||
# call model hook
|
||||
|
||||
Reference in new issue
Block a user