How to use multiple GPUs for training

[mikkker]

Hello, thank you for your excellent work. I have a question, how can I use multiple GPUs for training?

[Ayews]

Hello, thank you for your interest. Unfortunately, we did not implement multiple GPUS training in our code. You can implement the relevant parts yourself, mainly by modifying the base/framework_factory.py and train.py. However, it is worth mentioning that a single card with 24GB memory is sufficient for the training of this work, and the necessity of multiple GPUS training is not significant.