Fastai中的回调

问题描述 投票:0回答:1

我正在Fastai进行一个深度学习项目,并希望在其中使用带有ReduceLROnPlateauCallback的EarlyStoppingCallback。阅读callbacks.fastai,但努力理解如何实现这两者,但找不到任何相关示例。任何帮助,将不胜感激。

learn = cnn_learner(data, models.resnet50, metrics = [accuracy,quadratic_kappa])
learn.fit(50,2e-6)
callback pytorch fast-ai
1个回答
0
投票

我通常的操作方式是这种方式。...

首先创建学习者对象

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

    learn.unfreeze()

然后您在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')])
© www.soinside.com 2019 - 2024. All rights reserved.