Multiple GPU training via DistributedDataParallel

Rikorose / DeepFilterNet

Noise supression using deep filtering

https://huggingface.co/spaces/hshr/DeepFilterNet2

Other

2.11k stars 196 forks source link

Multiple GPU training via DistributedDataParallel #570

Open pkendrick opened 1 month ago

pkendrick commented 1 month ago

Hi there,

I have been trying to enable training on a a multi GPU machine, using the DistributedDataParallel method. I am realising that the current dataloader may need to be modified if it were to support this. Just wondered if anyone else had had a go at this also?

Cheers! Paul