mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-02-02 15:48:39 +09:00
parent d8bfb903db
commit 857843c872
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/aishell/ASR/conformer_ctc/.train.py.swp
View File

Binary file not shown.

3
egs/aishell/ASR/conformer_ctc/train.py
View File

@ -450,8 +450,6 @@ def compute_validation_loss(
# params.best_valid_loss = loss_value # params.best_valid_loss = loss_value
if params.cur_epoch >= 2 and loss_value < max(params.best_valid_losses.values()): if params.cur_epoch >= 2 and loss_value < max(params.best_valid_losses.values()):
remove_epoch = -1 remove_epoch = -1
params.best_valid_losses[params.cur_epoch] = loss_value
for k, v in params.best_valid_losses.items(): for k, v in params.best_valid_losses.items():
if v == max(params.best_valid_losses.values()): if v == max(params.best_valid_losses.values()):
remove_epoch = k remove_epoch = k
@ -459,6 +457,7 @@ def compute_validation_loss(
remove_filename = params.exp_dir / f"best-valid-loss_{remove_epoch}.pt" remove_filename = params.exp_dir / f"best-valid-loss_{remove_epoch}.pt"
if loss.device == 0: os.remove(remove_filename) if loss.device == 0: os.remove(remove_filename)
del params.best_valid_losses[k] del params.best_valid_losses[k]
params.best_valid_losses[params.cur_epoch] = loss_value
return tot_loss return tot_loss