search

LingxiaoYang2023 / DSG2024

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

About the training code. #1

Closed stefenmax closed 1 week ago

stefenmax commented 1 month ago

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

LingxiaoYang2023 commented 1 month 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).