facebookresearch/AudioDec

Multi-GPU training

jiaweiru opened this issue · 2 comments

Hello, thank you for your open source work. I have a question about this code, whether it can be trained using multiple GPUs. It seems that there is no option to set the GPU in the code.

bigpon commented

Hi,
since the computation cost is not high, we don't implement the multi-GPU training.
If you need a multi-GPU version, I would like to recommend you modify the code following the Parallel WaveGAN repo (https://github.com/kan-bayashi/ParallelWaveGAN).

Hi, since the computation cost is not high, we don't implement the multi-GPU training. If you need a multi-GPU version, I would like to recommend you modify the code following the Parallel WaveGAN repo (https://github.com/kan-bayashi/ParallelWaveGAN).

Thank you very much for your answer