facebookresearch / AudioDec

An Open-source Streaming High-fidelity Neural Audio Codec
Other
446 stars 21 forks source link

Multi-GPU training #3

Closed jiaweiru closed 1 year ago

jiaweiru commented 1 year ago

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 1 year ago

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).

jiaweiru commented 1 year ago

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