From fdeb9b88a79f5a2e297666f5eb0b9a3aec63ba76 Mon Sep 17 00:00:00 2001 From: WongLaw Date: Mon, 24 Apr 2023 03:01:10 +0000 Subject: [PATCH] VITS learning rate revised, test=tts --- examples/csmsc/vits/conf/default.yaml | 2 +- paddlespeech/t2s/exps/vits/train.py | 6 ++---- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/examples/csmsc/vits/conf/default.yaml b/examples/csmsc/vits/conf/default.yaml index e0572535d..a897569ac 100644 --- a/examples/csmsc/vits/conf/default.yaml +++ b/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 diff --git a/paddlespeech/t2s/exps/vits/train.py b/paddlespeech/t2s/exps/vits/train.py index 8e166beb7..fefa6a010 100644 --- a/paddlespeech/t2s/exps/vits/train.py +++ b/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()