Spaces:
Running
Running
File size: 1,464 Bytes
9d61c9b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
import torch
from torch.nn import Module
from models.config import VocoderModelConfig
from .multi_period_discriminator import MultiPeriodDiscriminator
from .multi_resolution_discriminator import MultiResolutionDiscriminator
class Discriminator(Module):
r"""Discriminator for the UnuvNet vocoder.
This class implements a discriminator that consists of a `MultiResolutionDiscriminator` and a `MultiPeriodDiscriminator`.
Args:
model_config (VocoderModelConfig): Model configuration object.
Attributes:
MRD (MultiResolutionDiscriminator): Multi-resolution discriminator instance.
MPD (MultiPeriodDiscriminator): Multi-period discriminator instance.
Methods:
forward(x): Computes the forward pass of the discriminator.
"""
def __init__(self, model_config: VocoderModelConfig):
super().__init__()
self.MRD = MultiResolutionDiscriminator(model_config=model_config)
self.MPD = MultiPeriodDiscriminator(model_config=model_config)
def forward(self, x: torch.Tensor) -> tuple[torch.Tensor, torch.Tensor]:
r"""Computes the forward pass of the discriminator.
Args:
x (torch.Tensor): Input tensor of shape [B, C, T].
Returns:
tuple(torch.Tensor, torch.Tensor): Tuple containing the output tensors of the `MultiResolutionDiscriminator` and `MultiPeriodDiscriminator` instances.
"""
return self.MRD(x), self.MPD(x)
|