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(