add extra_config keys into the default config of tacotron

This commit is contained in:
chenfeiyu 2021-04-30 14:27:08 +08:00
parent b88a0f90aa
commit 38831bf8b6
2 changed files with 16 additions and 11 deletions

View File

@ -32,6 +32,7 @@ _C.data = CN(
_C.model = CN(
dict(
vocab_size=37, # set this according to the frontend's vocab size
n_tones=None,
reduction_factor=1, # reduction factor
d_encoder=512, # embedding & encoder's internal size
encoder_conv_layers=3, # number of conv layer in tacotron2 encoder
@ -54,6 +55,7 @@ _C.model = CN(
p_decoder_dropout=
0.1, # droput probability of second rnn layer in decoder
p_postnet_dropout=0.5, # droput probability in decoder postnet
d_global_condition=None,
use_stop_token=
True, # wherther to use binary classifier to predict when to stop
use_guided_attention_loss=False, # whether to use guided attention loss

File diff suppressed because one or more lines are too long