提交 4ac7db18 编写于 作者: 小湉湉's avatar 小湉湉

init for all works in train.py when ngpu>1, test=tts

上级 d6edb62d
...@@ -62,9 +62,3 @@ Contents ...@@ -62,9 +62,3 @@ Contents
:caption: Acknowledgement :caption: Acknowledgement
asr/reference asr/reference
...@@ -162,7 +162,6 @@ def train_sp(args, config): ...@@ -162,7 +162,6 @@ def train_sp(args, config):
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), trigger=(1, 'epoch')) Snapshot(max_size=config.num_snapshots), trigger=(1, 'epoch'))
# print(trainer.extensions)
trainer.run() trainer.run()
......
...@@ -198,7 +198,6 @@ def train_sp(args, config): ...@@ -198,7 +198,6 @@ def train_sp(args, config):
Snapshot(max_size=config.num_snapshots), Snapshot(max_size=config.num_snapshots),
trigger=(config.save_interval_steps, 'iteration')) trigger=(config.save_interval_steps, 'iteration'))
# print(trainer.extensions.keys())
print("Trainer Done!") print("Trainer Done!")
trainer.run() trainer.run()
......
...@@ -157,7 +157,6 @@ def train_sp(args, config): ...@@ -157,7 +157,6 @@ def train_sp(args, config):
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), trigger=(1, 'epoch')) Snapshot(max_size=config.num_snapshots), trigger=(1, 'epoch'))
# print(trainer.extensions)
trainer.run() trainer.run()
......
...@@ -150,7 +150,6 @@ def train_sp(args, config): ...@@ -150,7 +150,6 @@ def train_sp(args, config):
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), trigger=(1, 'epoch')) Snapshot(max_size=config.num_snapshots), trigger=(1, 'epoch'))
# print(trainer.extensions)
trainer.run() trainer.run()
......
...@@ -137,7 +137,6 @@ def train_sp(args, config): ...@@ -137,7 +137,6 @@ def train_sp(args, config):
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), trigger=(1, 'epoch')) Snapshot(max_size=config.num_snapshots), trigger=(1, 'epoch'))
# print(trainer.extensions)
trainer.run() trainer.run()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册