search

ermongroup / SDEdit

PyTorch implementation for SDEdit: Image Synthesis and Editing with Stochastic Differential Equations
https://sde-image-editing.github.io/
MIT License
966 stars 90 forks source link

How to re-train a model with VP-SDE? #13

Open pokameng opened 1 year ago

pokameng commented 1 year ago

How to re-train a model with VP-SDE? @willieneis @junyanz @yang-song @jiamings @KellyYutongHe

chenlin9 commented 1 year ago

Hello, training a VP-SDE is the same as training a diffusion model. You may find the following repo useful:

  1. https://github.com/openai/guided-diffusion
  2. https://github.com/yang-song/score_sde