From e5feabb2326b3d07856e7cb2c9cdf9851680bf03 Mon Sep 17 00:00:00 2001 From: Mingshuang Luo <37799481+luomingshuang@users.noreply.github.com> Date: Tue, 12 Oct 2021 20:18:15 +0800 Subject: [PATCH] Update train.py --- egs/yesno/ASR/tdnn/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/yesno/ASR/tdnn/train.py b/egs/yesno/ASR/tdnn/train.py index 6cc511a28..d414962ca 100755 --- a/egs/yesno/ASR/tdnn/train.py +++ b/egs/yesno/ASR/tdnn/train.py @@ -416,7 +416,7 @@ def train_one_epoch( f"batch {batch_idx}, loss[{loss_info}], " f"tot_loss[{tot_loss}], batch size: {batch_size}" ) - if batch_idx % 10 == 0: + if batch_idx % params.log_interval == 0: if tb_writer is not None: loss_info.write_summary(