[Hackathon 7th] add missing keywords of aishell3/vits-vc (#3932)

* Update default.yaml

* Update train.py

* Update default.yaml
pull/3960/head
yinfan98 8 months ago committed by GitHub
parent c7d5b39ccf
commit 0ef3cfca31
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -230,15 +230,17 @@ 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.train_max_steps, "iteration"),
out=output_dir)
if dist.get_rank() == 0:
trainer.extend(
evaluator, trigger=(config.eval_interval_epochs, 'epoch'))
evaluator, trigger=(config.eval_interval_steps, 'iteration'))
trainer.extend(VisualDL(output_dir), trigger=(1, 'iteration'))
trainer.extend(
Snapshot(max_size=config.num_snapshots),
trigger=(config.save_interval_epochs, 'epoch'))
trigger=(config.save_interval_steps, 'iteration'))
print("Trainer Done!")
trainer.run()

Loading…
Cancel
Save