VITS learning rate revised, test=tts

pull/3186/head
WongLaw 1 year ago
parent fdeb9b88a7
commit 305375c310

@ -181,5 +181,5 @@ 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_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

@ -234,7 +234,7 @@ def train_sp(args, config):
if dist.get_rank() == 0:
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(
Snapshot(max_size=config.num_snapshots),

Loading…
Cancel
Save