speechbrain.nnet.loss.si_snr_loss 模块

# Authors:

  • Szu-Wei, Fu 2021

  • Mirco Ravanelli 2020

  • 萨穆埃莱·科内尔 2020

  • Hwidong Na 2020

  • 高岩 2020

  • Titouan Parcollet 2020

摘要

函数：

si_snr_loss

计算 si_snr 分数并返回 -1 * 该分数。

参考

speechbrain.nnet.loss.si_snr_loss.si_snr_loss(y_pred_batch, y_true_batch, lens, reduction='mean')[source]

计算si_snr分数并返回-1乘以该分数。

此函数可用作基于SGD更新的训练损失函数。

Parameters:

  • y_pred_batch (torch.Tensor) – 退化（增强）的波形。

  • y_true_batch (torch.Tensor) – 干净的（参考）波形。

  • lens (torch.Tensor) – 批次中波形的相对长度。

  • reduction (str) – 使用的减少类型（“mean”或“batch”）。

Return type:

计算si_snr损失。