提交 305375c3 编写于 作者: W WongLaw

VITS learning rate revised, test=tts

上级 fdeb9b88
...@@ -181,5 +181,5 @@ generator_first: False # whether to start updating generator first ...@@ -181,5 +181,5 @@ generator_first: False # whether to start updating generator first
num_snapshots: 10 # max number of snapshots to keep while training num_snapshots: 10 # max number of snapshots to keep while training
max_epoch: 1000 # Number of training epochs. max_epoch: 1000 # Number of training epochs.
save_interval_epochs: 1 # Interval epochs to save checkpoint. save_interval_epochs: 1 # Interval epochs to save checkpoint.
eval_interval_steps: 250 # Interval steps to evaluate the network. eval_interval_epochs: 1 # Interval steps to evaluate the network.
seed: 777 # random seed number seed: 777 # random seed number
...@@ -234,7 +234,7 @@ def train_sp(args, config): ...@@ -234,7 +234,7 @@ def train_sp(args, config):
if dist.get_rank() == 0: if dist.get_rank() == 0:
trainer.extend( trainer.extend(
evaluator, trigger=(config.eval_interval_steps, 'iteration')) evaluator, trigger=(config.eval_interval_epochs, 'epoch'))
trainer.extend(VisualDL(output_dir), trigger=(1, 'iteration')) trainer.extend(VisualDL(output_dir), trigger=(1, 'iteration'))
trainer.extend( trainer.extend(
Snapshot(max_size=config.num_snapshots), Snapshot(max_size=config.num_snapshots),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册