fit_flat_lin: Fit_flat_lin

Description Usage Arguments Value

View source: R/Learner_fits.R

Description

Fit 'self.model' for 'n_epoch' at flat 'start_lr' before 'curve_type' annealing to 'end_lr' with weight decay of 'wd' and callbacks 'cbs'.

Usage

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
fit_flat_lin(
  object,
  n_epochs = 100,
  n_epochs_decay = 100,
  start_lr = NULL,
  end_lr = 0,
  curve_type = "linear",
  wd = NULL,
  cbs = NULL,
  reset_opt = FALSE
)

Arguments

object

model / learner

n_epochs

number of epochs

n_epochs_decay

number of epochs with decay

start_lr

Desired starting learning rate, used for beginning pct of training.

end_lr

Desired end learning rate, training will conclude at this learning rate.

curve_type

Curve type for learning rate annealing. Options are 'linear', 'cosine', and 'exponential'.

wd

weight decay

cbs

callbacks

reset_opt

reset optimizer

Value

None


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