Fit_flat_lin

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`.

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