Code of multi gpus training of diffusion model

nihaomiao / CVPR23_LFDM

The pytorch implementation of our CVPR 2023 paper "Conditional Image-to-Video Generation with Latent Flow Diffusion Models"

BSD 2-Clause "Simplified" License

432 stars 43 forks source link

Code of multi gpus training of diffusion model #14

Closed ganchaofan0000 closed 11 months ago

ganchaofan0000 commented 11 months ago

I want to train the diffusion model with multi gpus. But there are not relevant codes.

nihaomiao commented 11 months ago

Hi, @ganchaofan0000, I just add the multi-GPU training codes for MHAD dataset. The training on the other datasets should be similar to MHAD. Please feel free to let me know if you have any other questions.

ganchaofan0000 commented 11 months ago

Thanks!