fumiama/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-06-08 12:00:49 +08:00
Code Activity

Changing 2333333 to latest

Browse Source 
Removing hard-coded "2333333" in latest saved checkpoint model. This make the code more logical and informative.
This commit is contained in:
Tegar Bangun Suganda
2024-07-30 10:00:44 +07:00
committed by 源文雨
parent 0884d286a9
commit 277eadca9a
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
infer/modules/train/train.py
View File

@@ -595,14 +595,14 @@ def train_and_evaluate(
optim_g, optim_g,
hps.train.learning_rate, hps.train.learning_rate,
epoch, epoch,
os.path.join(hps.model_dir, "G_{}.pth".format(2333333)), os.path.join(hps.model_dir, "G_latest.pth"),
) )
utils.save_checkpoint( utils.save_checkpoint(
net_d, net_d,
optim_d, optim_d,
hps.train.learning_rate, hps.train.learning_rate,
epoch, epoch,
os.path.join(hps.model_dir, "D_{}.pth".format(2333333)), os.path.join(hps.model_dir, "D_latest.pth"),
) )
if rank == 0 and hps.save_every_weights == "1": if rank == 0 and hps.save_every_weights == "1":
if hasattr(net_g, "module"): if hasattr(net_g, "module"):
@@ -644,7 +644,7 @@ def train_and_evaluate(
) )
) )
sleep(1) sleep(1)
os._exit(2333333) os._exit(0)
if __name__ == "__main__": if __name__ == "__main__":