From afc9f9c4138408a4db527e6a339b20dc63c3cef7 Mon Sep 17 00:00:00 2001 From: Yifan Yang Date: Fri, 2 Jun 2023 10:34:04 +0800 Subject: [PATCH] add multidataset --- icefall/rnn_lm/optim.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/icefall/rnn_lm/optim.py b/icefall/rnn_lm/optim.py index 0ceea7153..d26e1e1cc 100644 --- a/icefall/rnn_lm/optim.py +++ b/icefall/rnn_lm/optim.py @@ -179,6 +179,6 @@ class NewBobScheduler(LRScheduler): return { "base_lrs": self.base_lrs, "prev_metric": self.prev_metric, - "current_metric": current_metric, + "current_metric": self.current_metric, "current_patient": self.current_patient, }