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
b1711d1403
commit
41bfb30ce1
Binary file not shown.
Binary file not shown.
@ -1187,7 +1187,8 @@ def train_one_epoch(
|
||||
wb.log({"train/simple_loss": loss_info["simple_loss"]*numel})
|
||||
wb.log({"train/pruned_loss": loss_info["pruned_loss"]*numel})
|
||||
wb.log({"train/ctc_loss": loss_info["ctc_loss"]*numel})
|
||||
|
||||
|
||||
'''
|
||||
logging.info("Computing validation loss")
|
||||
valid_info = compute_validation_loss(
|
||||
params=params,
|
||||
@ -1216,7 +1217,7 @@ def train_one_epoch(
|
||||
wb.log({"valid/simple_loss": valid_info["simple_loss"]*numel})
|
||||
wb.log({"valid/pruned_loss": valid_info["pruned_loss"]*numel})
|
||||
wb.log({"valid/ctc_loss": valid_info["ctc_loss"]*numel})
|
||||
|
||||
'''
|
||||
loss_value = tot_loss["loss"] / tot_loss["utterances"]
|
||||
params.train_loss = loss_value
|
||||
if params.train_loss < params.best_train_loss:
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user