AWD_LSTM: AWD_LSTM

In fastai: Interface to 'fastai'

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.