espnet2.enh.loss.wrappers.fixed_order.FixedOrderSolver

source

class espnet2.enh.loss.wrappers.fixed_order.FixedOrderSolver(criterion: AbsEnhLoss, weight=1.0)

Bases: AbsLossWrapper

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

forward(ref, inf, others={})

An naive fixed-order solver

• Parameters:
  • ref (List *[*torch.Tensor ]) – [(batch, …), …] x n_spk
  • inf (List *[*torch.Tensor ]) – [(batch, …), …]
• Returns: (torch.Tensor): minimum loss with the best permutation stats: dict, for collecting training status others: reserved
• Return type: loss