srvk / eesen

The official repository of the Eesen project
http://arxiv.org/abs/1507.08240
Apache License 2.0
822 stars 342 forks source link

Support for multi-GPU #40

Closed fernand closed 8 years ago

fernand commented 8 years ago

I know that multi-GPU is not supported yet. I'm interested in knowing whether supporting that is straightforward or not, so that we know it's worth for us to work on it. We're interested using Eesen on more than 1000 hours of audio, so obviously that would be helpful (the 110 hours TEDLIUM dataset takes about 3 days with a single K40). Thanks!

gowayyed commented 8 years ago

I should be working on that soon. It is not straightfoward, but doable within a month or two if you know where everything is. Basically, you need to take the natural gradient code from kaldi and integrate it with eesen. The main idea to get parallelization to work is to scale the gradients accordingly, which the natural gradient does, as described in this paper (http://arxiv.org/pdf/1410.7455v4.pdf). This paper is already implemented in kaldi, so all we need to do is to migrate that to eesen.