Обратные вызовы в Фастае

Я работаю над проектом глубокого обучения в Фастае и хочу использовать в нем EarlyStoppingCallback с ReduceLROnPlateauCallback. Прочтите callbacks.fastai, но изо всех сил пытаетесь понять, как реализовать оба и не можете не найти подходящего примера. Любая помощь будет оценена по достоинству.

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

person Ayushman Koul    schedule 02.05.2020    source источник


Ответы (1)


Обычно я делаю так....

Сначала создайте объект учащегося

    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')])
person Oscar Rangel    schedule 06.05.2020