testadmin
/
Parakeet
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix priority in example/parallelwavegan

This commit is contained in:
chenfeiyu 2021-06-30 13:33:56 +08:00
parent ef51e1ab13
commit af26c1e389
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
examples/parallelwave_gan/baker/train.py
View File

@ -149,10 +149,10 @@ def train_sp(args, config):
output_dir = Path(args.output_dir)
checkpoint_dir = output_dir / "checkpoints"
if dist.get_rank() == 0:
with open(output_dir / "config.yaml", 'wt') as f:
f.write(config.dump(default_flow_style=None))
output_dir.mkdir(parents=True, exist_ok=True)
checkpoint_dir.mkdir(parents=True, exist_ok=True)
with open(output_dir / "config.yaml", 'wt') as f:
f.write(config.dump(default_flow_style=None))
updater = PWGUpdater(
models={
@ -192,9 +192,7 @@ def train_sp(args, config):
out=output_dir, )
trainer.extend(
evaluator,
trigger=(config.eval_interval_steps, 'iteration'),
priority=3)
evaluator, trigger=(config.eval_interval_steps, 'iteration'))
if dist.get_rank() == 0:
writer = LogWriter(str(trainer.out))
trainer.extend(VisualDL(writer), trigger=(1, 'iteration'))