search

ermongroup / SDEdit

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

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

Open pokameng opened 2 years ago

pokameng commented 2 years 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