fix bug

Mddct · Sep 5, 2024 · 1b2473b · 1b2473b
1 parent 306a26e
commit 1b2473b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/wenet/codec/vocos_low_latency/train.py b/wenet/codec/vocos_low_latency/train.py
@@ -322,6 +322,7 @@ def main():
     # TODO: args
     args = get_args()
     args.train_engine = 'torch_ddp'
+    args.dist_backend = 'nccl'
     _, _, rank = init_distributed(args)
 
     # init dataset