search

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.48k stars 479 forks source link

Any plans to lower CTC loss soon ? #2399

Open ASDen opened 4 years ago

ASDen commented 4 years ago

Hello,

I was eager to experiment with TPUs but was facing issues with CTC loss not lowered yet, so is this in your agenda soon? CTC is used very widely in OCR / ASR / many related sequence transduction problems.

also TF has a TPU implementation of CTC, so it shouldn't much hassle!

Thanks!

JackCaoG commented 4 years ago

@ASDen Thanks for reporting! We usually lower the op as user request so we will try to look at soon. It is a small team here so our bandwidth is limited, we will update this thread when we have more info.

usamec commented 4 years ago

I would also love to have this op lowered.

eedalong commented 3 years ago

I would also love to have this op lowered.

awasthiabhijeet commented 3 years ago

+1 for having the CTC_loss op lowered.

khursani8 commented 3 years ago

+1 this will help ASR/OCR research in future

JackCaoG commented 3 years ago

Sorry for the late response. We currently don't have the resource to implement this lowering given it is a bit more complicated comparing to other ops. We will prioritize this op given many user request and update here when we have a more concrete plan.

sagar-spkt commented 2 years ago

+1 for having the CTC_loss op lowered.

AlvinKimata commented 1 year ago

I would also love to have this op lowered.