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