run validation only at the end of epoch

This commit is contained in:
wangtiance 2023-01-19 17:32:49 +08:00
parent 623fe22ff1
commit 249b9300e9

View File

@ -395,7 +395,6 @@ def get_params() -> AttributeDict:
"batch_idx_train": 0,
"log_interval": 500,
"reset_interval": 200,
"valid_interval": 9000,
"warm_step": 5000,
"beam_size": 10,
"use_double_scores": True,
@ -927,7 +926,6 @@ def train_one_epoch(
"train/grad_scale", cur_grad_scale, params.batch_idx_train
)
if batch_idx > 0 and batch_idx % params.valid_interval == 0 and not params.print_diagnostics:
logging.info("Computing validation loss")
valid_info = compute_validation_loss(
params=params,