diff --git a/egs/aishell/ASR/conformer_ctc/.train.py.swp b/egs/aishell/ASR/conformer_ctc/.train.py.swp index e984b45c7..ead1a81a4 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 c8183da14..c1684d3e0 100755 --- a/egs/aishell/ASR/conformer_ctc/train.py +++ b/egs/aishell/ASR/conformer_ctc/train.py @@ -450,8 +450,6 @@ 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 @@ -459,6 +457,7 @@ def compute_validation_loss( remove_filename = params.exp_dir / f"best-valid-loss_{remove_epoch}.pt" if loss.device == 0: os.remove(remove_filename) del params.best_valid_losses[k] + params.best_valid_losses[params.cur_epoch] = loss_value return tot_loss