VITS learning rate revised, test=tts

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

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: