espnet2.enh.loss.criterions.tf_domain.FrequencyDomainAbsCoherence

source

class espnet2.enh.loss.criterions.tf_domain.FrequencyDomainAbsCoherence(compute_on_mask=False, mask_type=None, name=None, only_for_test=False, is_noise_loss=False, is_dereverb_loss=False)

Bases: FrequencyDomainLoss

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

property compute_on_mask : bool

forward(ref, inf) → Tensor

time-frequency absolute coherence loss.

Reference: : Independent Vector Analysis with Deep Neural Network Source Priors; Li et al 2020; https://arxiv.org/abs/2008.11273

• Parameters:
  • ref – (Batch, T, F) or (Batch, T, C, F)
  • inf – (Batch, T, F) or (Batch, T, C, F)
• Returns: (Batch,)
• Return type: loss

property mask_type : str