Resuming fit_one_cycle with model from SaveModelCallback?

fastai fastai dev
16

Try this:

class SkipToEpoch(Callback):
    def __init__(self,s_epoch): self.s_epoch = s_epoch
    def begin_train(self):  if self.epoch < self.s_epoch: raise CancelEpochException
    def begin_validate(self):  if self.epoch < self.s_epoch: raise CancelValidException

Then add it to your callbacks:
learn.fit_one_cycle(1,3e-3, cbs=cbs+SkipToEpoch(11))

of course, load the model before.

4 Likes