fit_flat_lin: Fit_flat_lin

View source: R/Learner_fits.R

fit_flat_linR Documentation

Fit_flat_lin

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

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 March 31, 2023, 11:41 p.m.