未验证 提交 312fc4e1 编写于 作者: H Hui Zhang 提交者: GitHub

Merge pull request #1766 from Jackwaterveg/fix

[ASR] ds2: add log_interval and  fix lr problem when resume training
......@@ -289,6 +289,7 @@ class Trainer():
float) else f"{v}"
msg += ","
msg = msg[:-1] # remove the last ","
if (batch_index + 1) % self.config.log_interval == 0:
logger.info(msg)
data_start_time = time.time()
except Exception as e:
......@@ -316,10 +317,10 @@ class Trainer():
self.visualizer.add_scalar(
tag='eval/lr', value=self.lr_scheduler(), step=self.epoch)
# after epoch
self.save(tag=self.epoch, infos={'val_loss': cv_loss})
# step lr every epoch
self.lr_scheduler.step()
# after epoch
self.save(tag=self.epoch, infos={'val_loss': cv_loss})
self.new_epoch()
def run(self):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册