espnet2.gan_svs.avocodo.avocodo.CoMBDBlock

source

class espnet2.gan_svs.avocodo.avocodo.CoMBDBlock(h_u: List[int], d_k: List[int], d_s: List[int], d_d: List[int], d_g: List[int], d_p: List[int], op_f: int, op_k: int, op_g: int, use_spectral_norm=False)

Bases: Module

CoMBD (Collaborative Multi-band Discriminator) block module

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

forward(x)

Forward pass through the CoMBD block.

• Parameters:x (Tensor) – Input tensor of shape (B, C_in, T_in).

• Returns: Tuple containing the output tensor of : shape (B, C_out, T_out)

  and a list of feature maps of shape (B, C, T) at each Conv1d layer.

• Return type: Tuple[Tensor, List[Tensor]]