提交 8af2eb07 编写于 作者: H Haoxin Ma

revise config

上级 68bcc469
...@@ -266,8 +266,8 @@ class Trainer(): ...@@ -266,8 +266,8 @@ class Trainer():
self.checkpoint_dir = checkpoint_dir self.checkpoint_dir = checkpoint_dir
self.checkpoint = KBestCheckpoint(max_size=self.config.training.max_epoch, self.checkpoint = KBestCheckpoint(max_size=self.config.training.checkpoint.kbest_n,
last_size=self.config.training.last_epoch) last_size=self.config.training.checkpoint.latest_n)
@mp_tools.rank_zero_only @mp_tools.rank_zero_only
def destory(self): def destory(self):
......
...@@ -49,8 +49,9 @@ training: ...@@ -49,8 +49,9 @@ training:
weight_decay: 1e-06 weight_decay: 1e-06
global_grad_clip: 5.0 global_grad_clip: 5.0
log_interval: 1 log_interval: 1
max_epoch: 3 checkpoint:
last_epoch: 2 kbest_n: 3
latest_n: 2
decoding: decoding:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册