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)
|
||||
|
||||
return torch.sum(loss)
|
||||
return loss
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user