language_model_learner: Language_model_learner

View source: R/text_models.R

language_model_learnerR Documentation

Language_model_learner

Description

Create a 'Learner' with a language model from 'dls' and 'arch'.

Usage

language_model_learner(
  dls,
  arch,
  config = NULL,
  drop_mult = 1,
  backwards = FALSE,
  pretrained = TRUE,
  pretrained_fnames = NULL,
  opt_func = Adam(),
  lr = 0.001,
  cbs = NULL,
  metrics = NULL,
  path = NULL,
  model_dir = "models",
  wd = NULL,
  wd_bn_bias = FALSE,
  train_bn = TRUE,
  moms = list(0.95, 0.85, 0.95),
  ...
)

Arguments

dls

dls

arch

arch

config

config

drop_mult

drop_mult

backwards

backwards

pretrained

pretrained

pretrained_fnames

pretrained_fnames

opt_func

opt_func

lr

lr

cbs

cbs

metrics

metrics

path

path

model_dir

model_dir

wd

wd

wd_bn_bias

wd_bn_bias

train_bn

train_bn

moms

moms

...

additional arguments

Value

None


fastai documentation built on March 21, 2022, 9:07 a.m.