Update s2_train.py (#1159)

1 year ago · e106a5ee88
parent 99f09c8bdc
commit e106a5ee88
1 changed files with 1 additions and 1 deletions
--- 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