From 414de3747c9ac9a7422dc45f0c945d135245de6f Mon Sep 17 00:00:00 2001 From: WongLaw Date: Thu, 20 Apr 2023 11:18:06 +0000 Subject: [PATCH] VITS learning rate revised, test=tts --- examples/csmsc/vits/conf/default.yaml | 2 +- paddlespeech/t2s/models/vits/vits_updater.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/examples/csmsc/vits/conf/default.yaml b/examples/csmsc/vits/conf/default.yaml index a2aef998..e0572535 100644 --- a/examples/csmsc/vits/conf/default.yaml +++ b/examples/csmsc/vits/conf/default.yaml @@ -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 diff --git a/paddlespeech/t2s/models/vits/vits_updater.py b/paddlespeech/t2s/models/vits/vits_updater.py index 9f8be680..7926bb6a 100644 --- a/paddlespeech/t2s/models/vits/vits_updater.py +++ b/paddlespeech/t2s/models/vits/vits_updater.py @@ -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: