mirror of
https://github.com/wassname/TTS.git
synced 2026-09-09 11:16:00 +08:00
refactor HifiGAN discriminator
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user