removed bad hook call

This commit is contained in:
William Falcon
2019-08-07 07:38:45 -04:00
parent a931ded310
commit d5fd16a478
+3 -3
View File
@@ -244,9 +244,6 @@ class Trainer(TrainerIO):
'''
raise ModuleNotFoundError(msg)
# restore training and model
self.restore_state_if_existing_checkpoint()
def restore_state_if_existing_checkpoint(self):
# restore trainer state and model if there is a weight for this experiment
last_epoch = -1
@@ -624,6 +621,9 @@ class Trainer(TrainerIO):
ref_model.trainer = self
ref_model.experiment = self.experiment
# restore training and model
self.restore_state_if_existing_checkpoint()
# run tiny validation to make sure program won't crash during val
_ = self.validate(model, self.val_dataloader, max_batches=self.nb_sanity_val_steps)