RNNModel.Config

Component: RNNModel

class RNNModel.Config

Bases: ConfigBase

All Attributes (including base classes)

encoder: LSTMSequenceEncoder.Config = LSTMSequenceEncoder.Config()

decoder: RNNDecoder.Config = RNNDecoder.Config()

Default JSON

{
    "encoder": {
        "embed_dim": 512,
        "hidden_dim": 512,
        "num_layers": 1,
        "dropout_in": 0.1,
        "dropout_out": 0.1,
        "bidirectional": false
    },
    "decoder": {
        "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
    }
}