fine_tune: Fine_tune

Description Usage Arguments Value

View source: R/Learner_fits.R

Description

Fine tune with 'freeze' for 'freeze_epochs' then with 'unfreeze' from 'epochs' using discriminative LR

Usage

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
fine_tune(
  object,
  epochs,
  base_lr = 0.002,
  freeze_epochs = 1,
  lr_mult = 100,
  pct_start = 0.3,
  div = 5,
  ...
)

Arguments

object

learner/model

epochs

epoch number

base_lr

base learning rate

freeze_epochs

freeze epochs number

lr_mult

learning rate multiply

pct_start

start percentage

div

divide

...

additional arguments

Value

None


fastai documentation built on July 28, 2021, 5:06 p.m.