How to multi-GPU training?

[yuuhaixia]

Hello, does the code support multi-GPU training? How should it be modified?

[jiaor17]

Yes! One could modifying the conf/train/default.yaml config by opening the accelerator: ddp comment and changing the gpus into the number of GPUs on the device.