mirror of
https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git
synced 2024-12-29 19:15:04 +08:00
Update train_nsf_sim_cache_sid_load_pretrain.py
This commit is contained in:
parent
9a855e0457
commit
6c7c1d933f
@ -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)
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user