diff --git a/espnet2/enh/loss/criterions/tf_domain.py b/espnet2/enh/loss/criterions/tf_domain.py index 6e20dede475..6a8647fe698 100644 --- a/espnet2/enh/loss/criterions/tf_domain.py +++ b/espnet2/enh/loss/criterions/tf_domain.py @@ -342,6 +342,7 @@ def name(self) -> str: def forward(self, ref, inf) -> torch.Tensor: """time-frequency absolute coherence loss. + Reference: Independent Vector Analysis with Deep Neural Network Source Priors; Li et al 2020; https://arxiv.org/abs/2008.11273 @@ -433,4 +434,4 @@ def forward(self, ref, inf) -> torch.Tensor: acc = acc.mean(dim=[1, 2]) self.stats = {"acc": acc.cpu() * 100} - return loss \ No newline at end of file + return loss diff --git a/espnet2/enh/separator/dpcl_e2e_separator.py b/espnet2/enh/separator/dpcl_e2e_separator.py index 311fec4daa3..35264c5c137 100644 --- a/espnet2/enh/separator/dpcl_e2e_separator.py +++ b/espnet2/enh/separator/dpcl_e2e_separator.py @@ -25,7 +25,7 @@ def __init__( dropout: float = 0.0, alpha: float = 5.0, max_iteration: int = 500, - threshold: float = 1.0e-05 + threshold: float = 1.0e-05, ): """Deep Clustering End-to-End Separator