How to show losses of all samples in a batch

SeanNaren / warp-ctc

Pytorch Bindings for warp-ctc

Apache License 2.0

756 stars 271 forks source link

How to show losses of all samples in a batch #169

Open ccx1997 opened 4 years ago

ccx1997 commented 4 years ago

I want the ctc loss_fn to return a tensor (with shape as [batch]) containing every loss in a batch, like the parameter reduction of CTCloss in Pytorch: ctcloss = torch.nn.CTCLoss(reduction='none') How can I do?