From 1b2473b46cbd18ac2469b8e3f14518ebc0bf1d43 Mon Sep 17 00:00:00 2001 From: Mddct Date: Fri, 6 Sep 2024 00:57:19 +0800 Subject: [PATCH] fix bug --- wenet/codec/vocos_low_latency/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/wenet/codec/vocos_low_latency/train.py b/wenet/codec/vocos_low_latency/train.py index 8dcb72cdb..3bc270118 100644 --- 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