espnet2.enh.loss.criterions.time_domain.TimeDomainMSE

source

class espnet2.enh.loss.criterions.time_domain.TimeDomainMSE(name=None, only_for_test=False, is_noise_loss=False, is_dereverb_loss=False)

Bases: TimeDomainLoss

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

forward(ref, inf) → Tensor

Time-domain MSE loss forward.

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