RNNDecoder.ConfigΒΆ

Component: RNNDecoder

class RNNDecoder.Config

Bases: RNNDecoderBase.Config

All Attributes (including base classes)

encoder_hidden_dim: int = 512
embed_dim: int = 512
hidden_dim: int = 512
out_embed_dim: int = 512
cell_type: str = 'lstm'
num_layers: int = 1
dropout_in: float = 0.1
dropout_out: float = 0.1
attention_type: str = 'dot'
attention_heads: int = 8
first_layer_attention: bool = False
averaging_encoder: bool = False

Default JSON

{
    "encoder_hidden_dim": 512,
    "embed_dim": 512,
    "hidden_dim": 512,
    "out_embed_dim": 512,
    "cell_type": "lstm",
    "num_layers": 1,
    "dropout_in": 0.1,
    "dropout_out": 0.1,
    "attention_type": "dot",
    "attention_heads": 8,
    "first_layer_attention": false,
    "averaging_encoder": false
}