From d7001bd9dd23ae2f98e49e85224fff78ddc139ee Mon Sep 17 00:00:00 2001
From: XXXXRT666 <157766680+XXXXRT666@users.noreply.github.com>
Date: Mon, 26 Aug 2024 02:13:22 +0800
Subject: [PATCH] Multi GPUs

---
 GPT_SoVITS/s2_train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/GPT_SoVITS/s2_train.py b/GPT_SoVITS/s2_train.py
index ff7f917..917f762 100644
--- a/GPT_SoVITS/s2_train.py
+++ b/GPT_SoVITS/s2_train.py
@@ -306,7 +306,7 @@ def train_and_evaluate(
         y_lengths,
         text,
         text_lengths,
-    ) in enumerate(tqdm(train_loader,position=rank+1,leave=(epoch==hps.train.epochs),postfix=f'epoch:{epoch}')):
+    ) in enumerate(tqdm(train_loader,position=rank+1,leave=(epoch==hps.train.epochs),postfix=f'epoch:{epoch}',disable=(rank!=0))):
         if torch.cuda.is_available():
             spec, spec_lengths = spec.cuda(rank, non_blocking=True), spec_lengths.cuda(
                 rank, non_blocking=True