Commit 8a6c12ff authored by pkufool's avatar pkufool
Browse files

Minor fixes

parent 90edd9ab
...@@ -129,7 +129,7 @@ boundary = torch.zeros((B, 4), dtype=torch.int64) ...@@ -129,7 +129,7 @@ boundary = torch.zeros((B, 4), dtype=torch.int64)
boundary[:, 2] = target_lengths boundary[:, 2] = target_lengths
boundary[:, 3] = num_frames boundary[:, 3] = num_frames
loss = fast_rnnt.rnnt_loss_simple( loss = fast_rnnt.rnnt_loss_smoothed(
lm=lm, lm=lm,
am=am, am=am,
symbols=symbols, symbols=symbols,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment