VITS learning rate revised, test=tts

pull/3186/head
WongLaw 2 years ago
parent c7d24ba42c
commit 414de3747c

@ -179,7 +179,7 @@ generator_first: False # whether to start updating generator first
# OTHER TRAINING SETTING #
##########################################################
num_snapshots: 10 # max number of snapshots to keep while training
train_max_steps: 350000 # Number of training steps. == total_iters / ngpus, total_iters = 1000000
max_epoch: 1000 # Number of training epochs.
save_interval_steps: 1000 # Interval steps to save checkpoint.
eval_interval_steps: 250 # Interval steps to evaluate the network.
seed: 777 # random seed number

@ -166,7 +166,8 @@ class VITSUpdater(StandardUpdater):
gen_loss.backward()
self.optimizer_g.step()
self.scheduler_g.step()
if self.state.iteration % self.updates_per_epoch == 0:
self.scheduler_g.step()
# reset cache
if self.model.reuse_cache_gen or not self.model.training:
@ -202,7 +203,8 @@ class VITSUpdater(StandardUpdater):
dis_loss.backward()
self.optimizer_d.step()
self.scheduler_d.step()
if self.state.iteration % self.updates_per_epoch == 0:
self.scheduler_d.step()
# reset cache
if self.model.reuse_cache_dis or not self.model.training:

Loading…
Cancel
Save