fit_flat_lin: Fit_flat_lin

In fastai: Interface to 'fastai'

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 June 22, 2024, 11:15 a.m.