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

linter fix

pull/10/head
Eren Golge 2019-11-12 13:51:22 +01:00
parent 5ed55f5bef
commit cd06a4c1e5
3 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
layers/common_layers.py
View File

@ -175,9 +175,6 @@ class GravesAttention(nn.Module):
phi_t = g_t * torch.exp(-0.5 * inv_sig_t * (mu_t_ - j)**2) phi_t = g_t * torch.exp(-0.5 * inv_sig_t * (mu_t_ - j)**2)
alpha_t = self.COEF * torch.sum(phi_t, 1) alpha_t = self.COEF * torch.sum(phi_t, 1)
if alpha_t.max() > 1e+3:
breakpoint()
# apply masking # apply masking
if mask is not None: if mask is not None:
alpha_t.data.masked_fill_(~mask, self._mask_value) alpha_t.data.masked_fill_(~mask, self._mask_value)

2
layers/tacotron2.py
View File

@ -254,7 +254,7 @@ class Decoder(nn.Module):
memories = torch.cat((memory, memories), dim=0) memories = torch.cat((memory, memories), dim=0)
memories = self._update_memory(memories) memories = self._update_memory(memories)
if speaker_embeddings is not None: if speaker_embeddings is not None:
memories = torch.cat([memories, speaker_embeddings], dim=-1) memories = torch.cat([memories, speaker_embeddings], dim=-1)
memories = self.prenet(memories) memories = self.prenet(memories)
self._init_states(inputs, mask=mask) self._init_states(inputs, mask=mask)

2
speaker_encoder/train.py
View File

@ -11,7 +11,7 @@ from TTS.speaker_encoder.dataset import MyDataset
from TTS.speaker_encoder.loss import GE2ELoss from TTS.speaker_encoder.loss import GE2ELoss
from TTS.speaker_encoder.model import SpeakerEncoder from TTS.speaker_encoder.model import SpeakerEncoder
from TTS.speaker_encoder.visual import plot_embeddings from TTS.speaker_encoder.visual import plot_embeddings
from TTS.speaker_encoder.generic_utils import save_best_model, save_checkpoint from TTS.speaker_encoder.generic_utils import save_best_model
from TTS.utils.audio import AudioProcessor from TTS.utils.audio import AudioProcessor
from TTS.utils.generic_utils import (NoamLR, check_update, copy_config_file, from TTS.utils.generic_utils import (NoamLR, check_update, copy_config_file,
count_parameters, count_parameters,