Basic class for a `Learner` in NLP.
TextLearner(
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
)
dls
model
alpha
beta
moms
loss_func
opt_func
lr
splitter
cbs
metrics
path
model_dir
wd
wd_bn_bias
train_bn
None