Обратные звонки в Фастай - PullRequest
0 голосов
/ 02 мая 2020

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

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

1 Ответ

0 голосов
/ 06 мая 2020

Как я обычно делаю, это так ....

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

    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')])
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...