mirror of https://github.com/coqui-ai/TTS.git
refactor HifiGAN discriminator
This commit is contained in:
parent
15f362d5b1
commit
773f1db6fa
|
@ -204,8 +204,6 @@ class HifiganDiscriminator(nn.Module):
|
|||
List[Tensor]: discriminator scores.
|
||||
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 += scores_
|
||||
feats += feats_
|
||||
return scores, feats
|
||||
scores, feats = self.mpd(x)
|
||||
scores_, feats_ = self.msd(x)
|
||||
return scores + scores_, feats + feats_
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# adopted from https://github.com/jik876/hifi-gan/blob/master/models.py
|
||||
import torch
|
||||
import torch.nn.functional as F
|
||||
import torch.nn as nn
|
||||
|
|
Loading…
Reference in New Issue