espnet2.gan_codec.abs_gan_codec.AbsGANCodec

source

class espnet2.gan_codec.abs_gan_codec.AbsGANCodec(*args, **kwargs)

Bases: ABC, Module

GAN-based Neural Codec model abstract class.

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

abstract decode(*args, **kwargs) → Tensor

Return decoded waveform from codecs.

abstract encode(*args, **kwargs) → Tensor

Return encoded codecs from waveform.

abstract forward(forward_generator, *args, **kwargs) → Dict[str, Tensor | Dict[str, Tensor] | int]

Return generator or discriminator loss.

abstract meta_info() → Dict[str, Any]

Return meta information of the codec.