espnet2.enh.loss.criterions.tf_domain.FrequencyDomainMSE

source

class espnet2.enh.loss.criterions.tf_domain.FrequencyDomainMSE(compute_on_mask=False, mask_type='IBM', 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 MSE loss.

• 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