Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make duration predictor dropout configurable

This commit is contained in:
Eren Gölge 2021-08-09 13:05:31 +00:00
parent 922142428c
commit f7a72552f1
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
TTS/tts/models/vits.py
View File

@ -79,6 +79,9 @@ class VitsArgs(Coqpit):
dropout_p_text_encoder (float):
Dropout rate of the text encoder. Defaults to 0.1.
dropout_p_duration_predictor (float):
Dropout rate of the duration predictor. Defaults to 0.1.
kernel_size_posterior_encoder (int):
Kernel size of the posterior encoder's WaveNet layers. Defaults to 5.
@ -174,6 +177,7 @@ class VitsArgs(Coqpit):
num_layers_text_encoder: int = 6
kernel_size_text_encoder: int = 3
dropout_p_text_encoder: int = 0.1
dropout_p_duration_predictor: int = 0.1
kernel_size_posterior_encoder: int = 5
dilation_rate_posterior_encoder: int = 1
num_layers_posterior_encoder: int = 16
@ -300,11 +304,11 @@ class Vits(BaseTTS):
if args.use_sdp:
self.duration_predictor = StochasticDurationPredictor(
args.hidden_channels, 192, 3, 0.5, 4, cond_channels=self.embedded_speaker_dim
args.hidden_channels, 192, 3, args.dropout_p_duration_predictor, 4, cond_channels=self.embedded_speaker_dim
)
else:
self.duration_predictor = DurationPredictor(
args.hidden_channels, 256, 3, 0.5, cond_channels=self.embedded_speaker_dim
args.hidden_channels, 256, 3, args.dropout_p_duration_predictor, cond_channels=self.embedded_speaker_dim
)
self.waveform_decoder = HifiganGenerator(