From 7879ecece73f5029a6a21339227c3b29cdbf849d Mon Sep 17 00:00:00 2001 From: hasangchun Date: Thu, 22 Jul 2021 03:25:00 +0900 Subject: [PATCH] Resolved #58 --- openspeech/models/contextnet/configurations.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/openspeech/models/contextnet/configurations.py b/openspeech/models/contextnet/configurations.py index 526d3b2..99ee77b 100644 --- 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(