VITS learning rate revised, test=tts

examples/csmsc/vits/conf/default.yaml

@@ -180,6 +180,6 @@ generator_first: False # whether to start updating generator first
 ##########################################################
 num_snapshots: 10            # max number of snapshots to keep while training
 max_epoch: 1000              # Number of training epochs.
-save_interval_steps: 1000    # Interval steps to save checkpoint.
+save_interval_epochs: 1      # Interval epochs to save checkpoint.
 eval_interval_steps: 250     # Interval steps to evaluate the network.
 seed: 777                    # random seed number

paddlespeech/t2s/exps/vits/train.py

@@ -230,9 +230,7 @@ def train_sp(args, config):
         output_dir=output_dir)
 
     trainer = Trainer(
-        updater,
-        stop_trigger=(config.max_epoch, 'epoch'),
-        out=output_dir)
+        updater, stop_trigger=(config.max_epoch, 'epoch'), out=output_dir)
 
     if dist.get_rank() == 0:
         trainer.extend(
@@ -240,7 +238,7 @@ def train_sp(args, config):
         trainer.extend(VisualDL(output_dir), trigger=(1, 'iteration'))
     trainer.extend(
         Snapshot(max_size=config.num_snapshots),
-        trigger=(config.save_interval_steps, 'iteration'))
+        trigger=(config.save_interval_epochs, 'epoch'))
 
     print("Trainer Done!")
     trainer.run()