bacow/GPT-SoVITS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修复resume epoch数识别错，每次resume都要都训一轮的问题

Browse Source 
修复resume epoch数识别错，每次resume都要都训一轮的问题
This commit is contained in:
RVC-Boss
2025-02-23 15:09:22 +08:00
committed by GitHub
parent 514fb692db
commit aa07216bba
3 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
GPT_SoVITS/s2_train_v3.py
View File

@@ -178,6 +178,7 @@ def run(rank, n_gpus, hps):
net_g,
optim_g,
)
epoch_str+=1
global_step = (epoch_str - 1) * len(train_loader)
# epoch_str = 1
# global_step = 0
@@ -188,7 +189,7 @@ def run(rank, n_gpus, hps):
if hps.train.pretrained_s2G != ""and hps.train.pretrained_s2G != None and os.path.exists(hps.train.pretrained_s2G):
if rank == 0:
logger.info("loaded pretrained %s" % hps.train.pretrained_s2G)
print(
print("loaded pretrained %s" % hps.train.pretrained_s2G,
net_g.module.load_state_dict(
torch.load(hps.train.pretrained_s2G, map_location="cpu")["weight"],
strict=False,
@@ -225,6 +226,7 @@ def run(rank, n_gpus, hps):
net_d=optim_d=scheduler_d=None
for epoch in range(epoch_str, hps.train.epochs + 1):
print("start training from epoch %s"%epoch)
if rank == 0:
train_and_evaluate(
rank,
@@ -254,6 +256,7 @@ def run(rank, n_gpus, hps):
)
scheduler_g.step()
# scheduler_d.step()
print("training done")
def train_and_evaluate(