testadmin
/
Parakeet
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #111 from iclementine/develop 

fix a config key error
This commit is contained in:
Feiyu Chan 2021-05-18 18:59:53 +08:00 committed by GitHub
parent 4f288a6d4f 3aa6aed0fb
commit 87a0cec0c7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/transformer_tts/config.py
View File

@ -39,7 +39,7 @@ _C.model = CN(
d_ffn=1024, # encoder_d_ffn & decoder_d_ffn d_ffn=1024, # encoder_d_ffn & decoder_d_ffn
encoder_layers=4, # number of transformer encoder layer encoder_layers=4, # number of transformer encoder layer
decoder_layers=4, # number of transformer decoder 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 d_postnet=256, # decoder postnet(cnn)'s internal channel
postnet_layers=5, # decoder postnet(cnn)'s layer postnet_layers=5, # decoder postnet(cnn)'s layer
postnet_kernel_size=5, # decoder postnet(cnn)'s kernel size postnet_kernel_size=5, # decoder postnet(cnn)'s kernel size

2
examples/transformer_tts/preprocess.py
View File

@ -38,7 +38,7 @@ def create_dataset(config, source_path, target_path, verbose=False):
processor = AudioProcessor( processor = AudioProcessor(
sample_rate=config.data.sample_rate, sample_rate=config.data.sample_rate,
n_fft=config.data.n_fft, n_fft=config.data.n_fft,
n_mels=config.data.d_mel, n_mels=config.data.n_mels,
win_length=config.data.win_length, win_length=config.data.win_length,
hop_length=config.data.hop_length, hop_length=config.data.hop_length,
fmax=config.data.fmax, fmax=config.data.fmax,

2
parakeet/models/transformer_tts.py
View File

@ -571,7 +571,7 @@ class TransformerTTS(nn.Layer):
frontend, frontend,
d_encoder=config.model.d_encoder, d_encoder=config.model.d_encoder,
d_decoder=config.model.d_decoder, d_decoder=config.model.d_decoder,
d_mel=config.data.d_mel, d_mel=config.data.n_mels,
n_heads=config.model.n_heads, n_heads=config.model.n_heads,
d_ffn=config.model.d_ffn, d_ffn=config.model.d_ffn,
encoder_layers=config.model.encoder_layers, encoder_layers=config.model.encoder_layers,