LMLearner — LMLearner • fastai

Add functionality to `TextLearner` when dealingwith a language model

Add functionality to `TextLearner` when dealing with a language model

LMLearner(
  dls,
  model,
  alpha = 2,
  beta = 1,
  moms = list(0.8, 0.7, 0.8),
  loss_func = NULL,
  opt_func = Adam(),
  lr = 0.001,
  splitter = trainable_params(),
  cbs = NULL,
  metrics = NULL,
  path = NULL,
  model_dir = "models",
  wd = NULL,
  wd_bn_bias = FALSE,
  train_bn = TRUE
)

LMLearner(
  dls,
  model,
  alpha = 2,
  beta = 1,
  moms = list(0.8, 0.7, 0.8),
  loss_func = NULL,
  opt_func = Adam(),
  lr = 0.001,
  splitter = trainable_params(),
  cbs = NULL,
  metrics = NULL,
  path = NULL,
  model_dir = "models",
  wd = NULL,
  wd_bn_bias = FALSE,
  train_bn = TRUE
)

Arguments

dls: dls
model: model
alpha: alpha
beta: beta
moms: moms
loss_func: loss_func
opt_func: opt_func
lr: lr
splitter: splitter
cbs: cbs
metrics: metrics
path: path
model_dir: model_dir
wd: wd
wd_bn_bias: wd_bn_bias
train_bn: train_bn

Value

text loader

None