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

View File

@ -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_

View File

@ -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