diff --git a/config.json b/config.json index c90482b9..713e3949 100644 --- a/config.json +++ b/config.json @@ -1,6 +1,6 @@ { "model_name": "TTS-phoneme", - "model_description": "Training with phonemes created by phonemizer.", + "model_description": "Training with phonemes, dropout rate 0.5", "audio":{ "audio_processor": "audio", // to use dictate different audio processors, if available. diff --git a/layers/tacotron.py b/layers/tacotron.py index d538b274..7b159c67 100644 --- a/layers/tacotron.py +++ b/layers/tacotron.py @@ -22,7 +22,7 @@ class Prenet(nn.Module): for (in_size, out_size) in zip(in_features, out_features) ]) self.relu = nn.ReLU() - self.dropout = nn.Dropout(0.1) + self.dropout = nn.Dropout(0.5) # self.init_layers() def init_layers(self):