Merge pull request #59 from openspeech-team/chun

Resolved #58
openspeech-team · Jul 21, 2021 · 4929bb2 · 4929bb2
2 parents 6b97e05 + 7879ece
commit 4929bb2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/openspeech/models/contextnet/configurations.py b/openspeech/models/contextnet/configurations.py
@@ -195,7 +195,7 @@ class ContextNetTransducerConfigs(OpenspeechDataclass):
     num_channels: int = field(
         default=256, metadata={"help": "The number of channels in the convolution filter"}
     )
-    hidden_dim: int = field(
+    decoder_hidden_state_dim: int = field(
         default=2048, metadata={"help": "The number of features in the decoder hidden state"}
     )
     encoder_dim: int = field(
@@ -204,7 +204,7 @@ class ContextNetTransducerConfigs(OpenspeechDataclass):
     decoder_output_dim: int = field(
         default=640, metadata={"help": "Dimension of decoder output vector"}
     )
-    dropout: float = field(
+    decoder_dropout_p: float = field(
         default=0.1, metadata={"help": "Dropout probability of decoder"}
     )
     rnn_type: str = field(