search

LingxiaoYang2023 / DSG2024

Official pytorch repository for “Guidance with Spherical Gaussian Constraint for Conditional Diffusion”
43 stars 2 forks source link

About the training code. #1

Closed stefenmax closed 5 months ago

stefenmax commented 5 months ago

Great work! Will you release the training codes/sripts? Thanks

LingxiaoYang2023 commented 5 months ago

The current code of DSG is based on DPS(https://github.com/DPS2022/diffusion-posterior-sampling), and DPS does not provide the train code. If you want to train the diffusion model, I recommend referring to the issues in DPS (e.g. https://github.com/DPS2022/diffusion-posterior-sampling/issues/3).