revise config

pull/680/head
Haoxin Ma 3 years ago
parent 68bcc46940
commit 8af2eb073a

@ -266,8 +266,8 @@ class Trainer():
self.checkpoint_dir = checkpoint_dir
self.checkpoint = KBestCheckpoint(max_size=self.config.training.max_epoch,
last_size=self.config.training.last_epoch)
self.checkpoint = KBestCheckpoint(max_size=self.config.training.checkpoint.kbest_n,
last_size=self.config.training.checkpoint.latest_n)
@mp_tools.rank_zero_only
def destory(self):

@ -49,8 +49,9 @@ training:
weight_decay: 1e-06
global_grad_clip: 5.0
log_interval: 1
max_epoch: 3
last_epoch: 2
checkpoint:
kbest_n: 3
latest_n: 2
decoding:

Loading…
Cancel
Save