Update infer.py

2025-09-04 06:34:20 +00:00 · 2023-12-04 10:00:51 +08:00 · 2023-12-04 10:00:51 +08:00 · e2cadd8380
commit e2cadd8380
parent ddecda0c9b
1 changed files with 6 additions and 1 deletions
--- a/egs/vctk/TTS/vits/infer.py
+++ b/egs/vctk/TTS/vits/infer.py
@ -247,7 +247,12 @@ def main():
    valid_cuts = vctk.valid_cuts()
    valid_dl = vctk.valid_dataloaders(valid_cuts)

-    infer_sets = {"test", test_dl, "valid", valid_dl}
+    infer_sets: Dict = {
+        "test",
+        test_dl,
+        "valid",
+        valid_dl,
+    }

    for subset, dl in infer_sets.items():
        save_wav_dir = params.res_dir / "wav" / subset