espnet2.enh.loss.criterions.tf_domain.FrequencyDomainLoss

source

class espnet2.enh.loss.criterions.tf_domain.FrequencyDomainLoss(name, only_for_test=False, is_noise_loss=False, is_dereverb_loss=False)

Bases: AbsEnhLoss, ABC

Base class for all frequence-domain Enhancement loss modules.

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

abstract property compute_on_mask : bool

create_mask_label(mix_spec, ref_spec, noise_spec=None)

property is_dereverb_loss : bool

property is_noise_loss : bool

abstract property mask_type : str

property name : str

property only_for_test : bool