LMLearner: LMLearner

In fastai: Interface to 'fastai'

LMLearner

Description

Add functionality to 'TextLearner' when dealingwith a language model

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

Usage

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

fastai documentation built on June 22, 2024, 11:15 a.m.