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-01-24 23:12:33 +09:00
parent b1711d1403
commit 41bfb30ce1
3 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/LJSpeech/ASR/.run_adapter.sh.swp
View File

Binary file not shown.

BIN
egs/LJSpeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp
View File

Binary file not shown.

3
egs/LJSpeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py
View File

@ -1188,6 +1188,7 @@ def train_one_epoch(
wb.log({"train/pruned_loss": loss_info["pruned_loss"]*numel}) wb.log({"train/pruned_loss": loss_info["pruned_loss"]*numel})
wb.log({"train/ctc_loss": loss_info["ctc_loss"]*numel}) wb.log({"train/ctc_loss": loss_info["ctc_loss"]*numel})
'''
logging.info("Computing validation loss") logging.info("Computing validation loss")
valid_info = compute_validation_loss( valid_info = compute_validation_loss(
params=params, params=params,
@ -1216,7 +1217,7 @@ def train_one_epoch(
wb.log({"valid/simple_loss": valid_info["simple_loss"]*numel}) wb.log({"valid/simple_loss": valid_info["simple_loss"]*numel})
wb.log({"valid/pruned_loss": valid_info["pruned_loss"]*numel}) wb.log({"valid/pruned_loss": valid_info["pruned_loss"]*numel})
wb.log({"valid/ctc_loss": valid_info["ctc_loss"]*numel}) wb.log({"valid/ctc_loss": valid_info["ctc_loss"]*numel})
'''
loss_value = tot_loss["loss"] / tot_loss["utterances"] loss_value = tot_loss["loss"] / tot_loss["utterances"]
params.train_loss = loss_value params.train_loss = loss_value
if params.train_loss < params.best_train_loss: if params.train_loss < params.best_train_loss: