Whether CTCLoss in pytorch support multi-gpu.

SeanNaren / deepspeech.pytorch

Speech Recognition using DeepSpeech2.

MIT License

2.1k stars 621 forks source link

Whether CTCLoss in pytorch support multi-gpu. #153

Closed SnailTyan closed 6 years ago

SnailTyan commented 6 years ago

@SeanNaren I want to know whether CTCLoss in pytorch support multi-gpu. When I use CTCLoss trainig crnn model, I find CTCLoss can't support multi-gpu, is it right?

SeanNaren commented 6 years ago

I'm not sure if the actual loss can be put inside the data parallel table, but the current implementation in the master repo should work! What error are you running into?

SeanNaren commented 6 years ago

Going to close this for now...