Update test_run in wavernn and wavegrad

2026-09-09 11:16:00 +08:00 · 2022-02-25 11:31:56 +01:00
parent bf540f4323
commit e0f9be76c0
1 changed files with 3 additions and 1 deletions
@@ -307,7 +307,9 @@ class Wavegrad(BaseVocoder):
        y = y.unsqueeze(1)
        return {"input": m, "waveform": y}

-    def get_data_loader(self, config: Coqpit, assets: Dict, is_eval: True, samples: List, verbose: bool, num_gpus: int):
+    def get_data_loader(
+        self, config: Coqpit, assets: Dict, is_eval: True, samples: List, verbose: bool, num_gpus: int
+    ):
        ap = assets["audio_processor"]
        dataset = WaveGradDataset(
            ap=ap,