mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
just remove unnecessary torch.sum
This commit is contained in:
parent
2332ba312d
commit
e558be1cb7
@ -122,4 +122,4 @@ class Transducer(nn.Module):
|
|||||||
|
|
||||||
loss = k2.rnnt_loss(logits, y_padded, blank_id, boundary)
|
loss = k2.rnnt_loss(logits, y_padded, blank_id, boundary)
|
||||||
|
|
||||||
return torch.sum(loss)
|
return loss
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user