AWD_LSTM — AWD_LSTM • fastai

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

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