AWD_LSTM: AWD_LSTM

View source: R/text_models.R

AWD_LSTMR Documentation

AWD_LSTM

Description

AWD-LSTM inspired by https://arxiv.org/abs/1708.02182

Usage

AWD_LSTM(
  vocab_sz,
  emb_sz,
  n_hid,
  n_layers,
  pad_token = 1,
  hidden_p = 0.2,
  input_p = 0.6,
  embed_p = 0.1,
  weight_p = 0.5,
  bidir = FALSE
)

Arguments

vocab_sz

vocab_sz

emb_sz

emb_sz

n_hid

n_hid

n_layers

n_layers

pad_token

pad_token

hidden_p

hidden_p

input_p

input_p

embed_p

embed_p

weight_p

weight_p

bidir

bidir

Value

None


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