diff --git a/egs/aishell/ASR/conformer_ctc/.train.py.swp b/egs/aishell/ASR/conformer_ctc/.train.py.swp index 5825d7a7d..e984b45c7 100644 Binary files a/egs/aishell/ASR/conformer_ctc/.train.py.swp and b/egs/aishell/ASR/conformer_ctc/.train.py.swp differ diff --git a/egs/aishell/ASR/conformer_ctc/train.py b/egs/aishell/ASR/conformer_ctc/train.py index 82117abe2..c8183da14 100755 --- a/egs/aishell/ASR/conformer_ctc/train.py +++ b/egs/aishell/ASR/conformer_ctc/train.py @@ -450,10 +450,11 @@ def compute_validation_loss( # params.best_valid_loss = loss_value if params.cur_epoch >= 2 and loss_value < max(params.best_valid_losses.values()): remove_epoch = -1 + params.best_valid_losses[params.cur_epoch] = loss_value + for k, v in params.best_valid_losses.items(): if v == max(params.best_valid_losses.values()): remove_epoch = k - params.best_valid_losses[params.cur_epoch] = loss_value remove_filename = params.exp_dir / f"best-valid-loss_{remove_epoch}.pt" if loss.device == 0: os.remove(remove_filename)