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

refactor HifiGAN discriminator

This commit is contained in:
Eren Gölge 2021-04-08 11:14:10 +02:00
parent 15f362d5b1
commit 773f1db6fa
2 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
TTS/vocoder/models/hifigan_discriminator.py
View File

@ -204,8 +204,6 @@ class HifiganDiscriminator(nn.Module):
List[Tensor]: discriminator scores. List[Tensor]: discriminator scores.
List[List[Tensor]]: list of list of features from each layers of each discriminator. List[List[Tensor]]: list of list of features from each layers of each discriminator.
""" """
scores, feats = self.msd(x) scores, feats = self.mpd(x)
scores_, feats_ = self.mpd(x) scores_, feats_ = self.msd(x)
scores += scores_ return scores + scores_, feats + feats_
feats += feats_
return scores, feats

1
TTS/vocoder/models/hifigan_generator.py
View File

@ -1,3 +1,4 @@
# adopted from https://github.com/jik876/hifi-gan/blob/master/models.py
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
import torch.nn as nn import torch.nn as nn