From 391432b35644a9cdf93b55351b1950d666eea256 Mon Sep 17 00:00:00 2001
From: Mingshuang Luo <37799481+luomingshuang@users.noreply.github.com>
Date: Tue, 12 Oct 2021 21:30:31 +0800
Subject: [PATCH] Update train.py ("10"--->"params.log_interval") (#76)

* Update train.py

* Update train.py

* Update train.py
---
 egs/librispeech/ASR/conformer_ctc/train.py | 2 +-
 egs/librispeech/ASR/tdnn_lstm_ctc/train.py | 2 +-
 egs/yesno/ASR/tdnn/train.py                | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/egs/librispeech/ASR/conformer_ctc/train.py b/egs/librispeech/ASR/conformer_ctc/train.py
index 3e1049fbf..5554aaa7c 100755
--- a/egs/librispeech/ASR/conformer_ctc/train.py
+++ b/egs/librispeech/ASR/conformer_ctc/train.py
@@ -485,7 +485,7 @@ def train_one_epoch(
                 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(
diff --git a/egs/librispeech/ASR/tdnn_lstm_ctc/train.py b/egs/librispeech/ASR/tdnn_lstm_ctc/train.py
index 51a486e07..4a8574019 100755
--- a/egs/librispeech/ASR/tdnn_lstm_ctc/train.py
+++ b/egs/librispeech/ASR/tdnn_lstm_ctc/train.py
@@ -435,7 +435,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(
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(