tcapelle / cloud_diffusion

Diffusion on the Clouds: Short-term solar energy forecasting with Diffusion Models

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Distributed / Parallel version

miquel-espinosa opened this issue · comments

Hi! Thanks for this project.
Is there any distributed or parallel version, to run the code in multiple GPUs?
Thanks

Not right now, but I would gladly review a PR that adds this feature to the MiniTrainer class. You could use Accelerate or just vanilla DDP.