Add files via upload

2025-02-23 20:26:18 +08:00
parent aa07216bba
commit fbb9f21e53
3 changed files with 6 additions and 6 deletions
--- a/GPT_SoVITS/s2_train.py
+++ b/GPT_SoVITS/s2_train.py
@@ -251,8 +251,8 @@ def run(rank, n_gpus, hps):

    scaler = GradScaler(enabled=hps.train.fp16_run)

+    print("start training from epoch %s" % epoch_str)
    for epoch in range(epoch_str, hps.train.epochs + 1):
-        print("start training from epoch %s"%epoch)
        if rank == 0:
            train_and_evaluate(
                rank,
@@ -282,7 +282,7 @@ def run(rank, n_gpus, hps):
            )
        scheduler_g.step()
        scheduler_d.step()
-        print("training done")
+    print("training done")


 def train_and_evaluate(