espnet2.gan_svs.avocodo.avocodo.CoMBD

source

class espnet2.gan_svs.avocodo.avocodo.CoMBD(h, pqmf_list=None, use_spectral_norm=False)

Bases: Module

CoMBD (Collaborative Multi-band Discriminator) module

from from https://arxiv.org/abs/2206.13404

Initializes internal Module state, shared by both nn.Module and ScriptModule.

forward(ys, ys_hat)

Forward CoMBD.

• Parameters:
  • ys (List *[*Tensor ]) – List of ground truth signals of shape (B, 1, T).
  • ys_hat (List *[*Tensor ]) – List of predicted signals of shape (B, 1, T).
• Returns: Tuple containing the list of output tensors of shape (B, C_out, T_out) for real and fake, respectively, and the list of feature maps of shape (B, C, T) at each Conv1d layer for real and fake, respectively.
• Return type: Tuple[List[Tensor], List[Tensor], List[List[Tensor]], List[List[Tensor]]]