search

jpuigcerver / pytorch-baidu-ctc

PyTorch bindinga for Baidu's Warp-CTC
Apache License 2.0
59 stars 11 forks source link

documentation #3

Open phtephanx opened 5 years ago

phtephanx commented 5 years ago

Thanks for the bindings!

I simply wanted to point out that you could add to the documentation:

WenmuZhou commented 5 years ago

can I use this with the same input as torch.nn.CTCloss?

jpuigcerver commented 5 years ago

Hi @WenmuZhou,

Not exactly. You need to make sure that you place the tensors in the appropriate devices.

Take a look at this piece of code, were I use both implementations of the CTC loss: https://github.com/jpuigcerver/PyLaia/blob/41d2cc41d742e7ab336393fde8f56585ff49ee52/laia/losses/ctc_loss.py#L350

WenmuZhou commented 5 years ago

tks@jpuigcerver

WeihongM commented 5 years ago

Hi, @jpuigcerver ask for one question: the input pred needs the log_softmax output?