From 96e0d92fb79e3220b5a678d1561833ce9edcc3dd Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Mon, 3 Oct 2022 18:24:00 +0800 Subject: [PATCH] Compute valid loss on batch 0. --- egs/librispeech/ASR/pruned_transducer_stateless7/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py index aa345dd84..e1090f6fd 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py @@ -838,7 +838,7 @@ def train_one_epoch( tb_writer, "train/tot_", params.batch_idx_train ) - if batch_idx > 0 and batch_idx % params.valid_interval == 0: + if batch_idx % params.valid_interval == 0: logging.info("Computing validation loss") valid_info = compute_validation_loss( params=params,