search

XY-boy / EDiffSR

[IEEE TGRS 2024] EDiffSR: An Efficient Diffusion Probabilistic Model for Remote Sensing Image Super-Resolution
103 stars 6 forks source link

Performing Distributed Training #7

Open dreamwish1998 opened 5 months ago

dreamwish1998 commented 5 months ago

How do we perform distributed training in this project? or how to modify the code for distributed training? Thank you very much!!!

XY-boy commented 5 months ago

How do we perform distributed training in this project? or how to modify the code for distributed training? Thank you very much!!!

I train it on a single GPU. Maybe you can set gpu_ids in the setting.yml to train it on multiple GPUs, e.g., gpu_ids: [0, 1, 2, 3]