32k config

Mddct · Sep 5, 2024 · 4403205 · 4403205
1 parent d16c1c5
commit 4403205
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wenet/codec/vocos_low_latency/train.py b/wenet/codec/vocos_low_latency/train.py
@@ -335,7 +335,7 @@ def main():
                                  seed=2024)
     device = torch.device(args.device)
     # init model
-    model_config = vocos_config()
+    model_config = vocos_config(input_size=128, n_fft=2048, hop_length=640)
     model = Vocosv1(model_config)
     multiperioddisc = MultiPeriodDiscriminator()
     multiresddisc = MultiResolutionDiscriminator()