未验证 提交 718c849f 编写于 作者: H Hui Zhang 提交者: GitHub

Merge pull request #1445 from yt605155624/update_train

[TTS]init for all works in train.py when ngpu>1
......@@ -62,9 +62,3 @@ Contents
:caption: Acknowledgement
asr/reference
......@@ -162,7 +162,6 @@ def train_sp(args, config):
trainer.extend(VisualDL(output_dir), trigger=(1, "iteration"))
trainer.extend(
Snapshot(max_size=config.num_snapshots), trigger=(1, 'epoch'))
# print(trainer.extensions)
trainer.run()
......
......@@ -198,7 +198,6 @@ def train_sp(args, config):
Snapshot(max_size=config.num_snapshots),
trigger=(config.save_interval_steps, 'iteration'))
# print(trainer.extensions.keys())
print("Trainer Done!")
trainer.run()
......
......@@ -157,7 +157,6 @@ def train_sp(args, config):
trainer.extend(VisualDL(output_dir), trigger=(1, "iteration"))
trainer.extend(
Snapshot(max_size=config.num_snapshots), trigger=(1, 'epoch'))
# print(trainer.extensions)
trainer.run()
......
......@@ -150,7 +150,6 @@ def train_sp(args, config):
trainer.extend(VisualDL(output_dir), trigger=(1, "iteration"))
trainer.extend(
Snapshot(max_size=config.num_snapshots), trigger=(1, 'epoch'))
# print(trainer.extensions)
trainer.run()
......
......@@ -137,7 +137,6 @@ def train_sp(args, config):
trainer.extend(VisualDL(output_dir), trigger=(1, "iteration"))
trainer.extend(
Snapshot(max_size=config.num_snapshots), trigger=(1, 'epoch'))
# print(trainer.extensions)
trainer.run()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册