OPENGIT/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git synced 2025-05-06 20:01:37 +08:00
Code Issues Packages Projects Releases Wiki 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 GitHub
parent 85829da3a0
commit 7abb76d552
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -583,14 +583,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"):
@ -632,7 +632,7 @@ def train_and_evaluate(
) )
) )
sleep(1) sleep(1)
os._exit(2333333) os._exit(0)
if __name__ == "__main__": if __name__ == "__main__":