mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
8deeece377
commit
cb0aa1247d
Binary file not shown.
@ -1089,7 +1089,7 @@ def train_one_epoch(
|
||||
wb.log({"valid/pruned_loss": valid_info["pruned_loss"]*numel})
|
||||
wb.log({"valid/loss": valid_info["loss"]*numel})
|
||||
|
||||
loss_value = tot_loss["loss"] / tot_loss["frames"]
|
||||
loss_value = tot_loss["loss"] / tot_loss["utterances"]
|
||||
params.train_loss = loss_value
|
||||
if params.train_loss < params.best_train_loss:
|
||||
params.best_train_epoch = params.cur_epoch
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user