Don't compute validation if printing diagnostics.

This commit is contained in:
Daniel Povey 2022-10-07 14:03:17 +08:00
parent bd325e8769
commit 314f2381e2

View File

@ -858,7 +858,7 @@ def train_one_epoch(
tb_writer, "train/tot_", params.batch_idx_train tb_writer, "train/tot_", params.batch_idx_train
) )
if batch_idx % params.valid_interval == 0: if batch_idx % params.valid_interval == 0 and not params.print_diagnostics:
logging.info("Computing validation loss") logging.info("Computing validation loss")
valid_info = compute_validation_loss( valid_info = compute_validation_loss(
params=params, params=params,