espnet2.enh.loss.wrappers.dpcl_solver.DPCLSolver

source

class espnet2.enh.loss.wrappers.dpcl_solver.DPCLSolver(criterion: AbsEnhLoss, weight=1.0)

Bases: AbsLossWrapper

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

forward(ref, inf, others={})

A naive DPCL solver

• Parameters:
  • ref (List *[*torch.Tensor ]) – [(batch, …), …] x n_spk
  • inf (List *[*torch.Tensor ]) – [(batch, …), …]
  • others (List) – other data included in this solver e.g. “tf_embedding” learned embedding of all T-F bins (B, T * F, D)
• Returns: (torch.Tensor): minimum loss with the best permutation stats: (dict), for collecting training status others: reserved
• Return type: loss