Fix format

chintu619 · Apr 27, 2022 · 0d3be31 · 0d3be31
1 parent b24d108
commit 0d3be31
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
+        return loss
diff --git 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