From e106a5ee8848a65b5a8f242c57a50f4d90d5cf0c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BD=AD=E9=9C=87=E4=B8=9C?= <zhendong.peng@qq.com>
Date: Fri, 7 Jun 2024 22:03:27 +0800
Subject: [PATCH] Update s2_train.py (#1159)

---
 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 4f0ca4c..ddbe2ab 100644
--- a/GPT_SoVITS/s2_train.py
+++ b/GPT_SoVITS/s2_train.py
@@ -305,7 +305,7 @@ def train_and_evaluate(
         y_lengths,
         text,
         text_lengths,
-    ) in tqdm(enumerate(train_loader)):
+    ) in enumerate(tqdm(train_loader)):
         if torch.cuda.is_available():
             spec, spec_lengths = spec.cuda(rank, non_blocking=True), spec_lengths.cuda(
                 rank, non_blocking=True