fix a config key error
This commit is contained in:
parent
4f288a6d4f
commit
3aa6aed0fb
|
@ -39,7 +39,7 @@ _C.model = CN(
|
|||
d_ffn=1024, # encoder_d_ffn & decoder_d_ffn
|
||||
encoder_layers=4, # number of transformer encoder layer
|
||||
decoder_layers=4, # number of transformer decoder layer
|
||||
d_prenet=256, # decprenet's hidden size (d_mel=>d_prenet=>d_decoder)
|
||||
d_prenet=256, # decoder prenet's hidden size (n_mels=>d_prenet=>d_decoder)
|
||||
d_postnet=256, # decoder postnet(cnn)'s internal channel
|
||||
postnet_layers=5, # decoder postnet(cnn)'s layer
|
||||
postnet_kernel_size=5, # decoder postnet(cnn)'s kernel size
|
||||
|
|
|
@ -38,7 +38,7 @@ def create_dataset(config, source_path, target_path, verbose=False):
|
|||
processor = AudioProcessor(
|
||||
sample_rate=config.data.sample_rate,
|
||||
n_fft=config.data.n_fft,
|
||||
n_mels=config.data.d_mel,
|
||||
n_mels=config.data.n_mels,
|
||||
win_length=config.data.win_length,
|
||||
hop_length=config.data.hop_length,
|
||||
fmax=config.data.fmax,
|
||||
|
|
|
@ -571,7 +571,7 @@ class TransformerTTS(nn.Layer):
|
|||
frontend,
|
||||
d_encoder=config.model.d_encoder,
|
||||
d_decoder=config.model.d_decoder,
|
||||
d_mel=config.data.d_mel,
|
||||
d_mel=config.data.n_mels,
|
||||
n_heads=config.model.n_heads,
|
||||
d_ffn=config.model.d_ffn,
|
||||
encoder_layers=config.model.encoder_layers,
|
||||
|
|
Loading…
Reference in New Issue