mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-10 02:22:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove logging code that broke with newer Lhotse; fix bug with pruned_loss

Browse Source
This commit is contained in:
Daniel Povey 2022-03-19 14:01:45 +08:00
parent 8cff994cd7
commit 0ee2404ff0
1 changed files with 2 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
egs/librispeech/ASR/pruned_transducer_stateless2/train.py
View File

@ -450,9 +450,8 @@ def compute_loss(
lm_scale=params.lm_scale,
warmup_mode=warmup_mode,
)
loss = params.simple_loss_scale * simple_loss
if not warmup_mode:
loss = loss + (pruned_loss * 0.01 if warmup_mode else pruned_loss)
loss = (params.simple_loss_scale * simple_loss +
(pruned_loss * 0.01 if warmup_mode else pruned_loss))
assert loss.requires_grad == is_training
@ -687,18 +686,8 @@ def run(rank, world_size, args):
# Keep only utterances with duration between 1 second and 20 seconds
return 1.0 <= c.duration <= 20.0
num_in_total = len(train_cuts)
train_cuts = train_cuts.filter(remove_short_and_long_utt)
num_left = len(train_cuts)
num_removed = num_in_total - num_left
removed_percent = num_removed / num_in_total * 100
logging.info(f"Before removing short and long utterances: {num_in_total}")
logging.info(f"After removing short and long utterances: {num_left}")
logging.info(f"Removed {num_removed} utterances ({removed_percent:.5f}%)")
train_dl = librispeech.train_dataloaders(train_cuts)
valid_cuts = librispeech.dev_clean_cuts()