Update train_nsf_sim_cache_sid_load_pretrain.py

This commit is contained in:
liujing04 2023-04-10 11:08:26 +08:00 committed by GitHub
parent 9a855e0457
commit 6c7c1d933f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -501,13 +501,12 @@ def train_and_evaluate(
if rank == 0:
logger.info("====> Epoch: {}".format(epoch))
if(epoch>=hps.total_epoch):
if rank == 0:
logger.info("Training is done. The program is closed.")
from process_ckpt import savee#def savee(ckpt,sr,if_f0,name,epoch):
if hasattr(net_g, 'module'):ckpt = net_g.module.state_dict()
else:ckpt = net_g.state_dict()
print("saving final ckpt:",savee(ckpt,hps.sample_rate,hps.if_f0,hps.name,epoch))
if(epoch>=hps.total_epoch and rank == 0):
logger.info("Training is done. The program is closed.")
from process_ckpt import savee#def savee(ckpt,sr,if_f0,name,epoch):
if hasattr(net_g, 'module'):ckpt = net_g.module.state_dict()
else:ckpt = net_g.state_dict()
logger.info("saving final ckpt:%s"%(savee(ckpt,hps.sample_rate,hps.if_f0,hps.name,epoch)))
os._exit(2333333)