add extra_config keys into the default config of tacotron
This commit is contained in:
parent
b88a0f90aa
commit
38831bf8b6
|
@ -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
Loading…
Reference in New Issue