removed bad hook call

This commit is contained in:
William Falcon 2019-08-07 07:38:45 -04:00
parent a931ded310
commit d5fd16a478
1 changed files with 3 additions and 3 deletions

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)