RNNDecoderBase.ConfigΒΆ
Component: RNNDecoderBase
-
class
RNNDecoderBase.
Config
[source] Bases:
ConfigBase
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
- Subclasses
RNNDecoder.Config
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
}