Configs RNNPredictionNetwork.
| 703 | |
| 704 | @config_class |
| 705 | class Config(BaseLayer.Config): |
| 706 | """Configs RNNPredictionNetwork.""" |
| 707 | |
| 708 | # Vocab size. |
| 709 | vocab_size: Required[int] = REQUIRED |
| 710 | # The embedding dim. |
| 711 | emb_dim: Required[int] = REQUIRED |
| 712 | # The output dim. |
| 713 | output_dim: Required[int] = REQUIRED |
| 714 | |
| 715 | # Embedding lookup layer. |
| 716 | embedding: Embedding.Config = Embedding.default_config() |
| 717 | # RNN cell of the internal LM. Defaults to a 1 layer LSTM. |
| 718 | rnn_cell: BaseRNNCell.Config = LSTMCell.default_config() |
| 719 | |
| 720 | def __init__(self, cfg: Config, *, parent: Optional[Module]): |
| 721 | super().__init__(cfg, parent=parent) |
nothing calls this directly
no test coverage detected