language_model_learner: Language_model_learner

Description Usage Arguments Value

View source: R/text_models.R

Description

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

Usage

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
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 July 28, 2021, 5:06 p.m.