Search code examples
callbackpytorchfast-ai

Callbacks in Fastai


I am working on a deep learning project in Fastai and wish to use EarlyStoppingCallback with ReduceLROnPlateauCallback in it. Read callbacks.fastai but struggling to understand how to implement both and couldn't find any relevant example. Any help would be appreciated.

learn = cnn_learner(data, models.resnet50, metrics = [accuracy,quadratic_kappa])
learn.fit(50,2e-6)

Solution

  • The way I normally do it is this way....

    First create the learner Object

        learn = Learner(data, 
                        model, 
                        loss_func=...., 
                        opt_func=...., 
                        metrics=..... )
    
        learn.unfreeze()
    

    Then you call any callbacks on your fit_one_cycle

        learn.fit_one_cycle(16, 
                            max_lr=1e-3, 
                            div_factor=100,
                            pct_start=0.0, 
                            callbacks = [SaveModelCallback(learn,
                                         name=f'model',monitor='kappa_score')])