search

AlexiaJM / score_sde_fast_sampling

Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper
105 stars 8 forks source link

pytorch implementation and ODE #2

Closed gwang-kim closed 3 years ago

gwang-kim commented 3 years ago

Hi, thanks for your great works.

Do you have any plan to implement the codes with Pytorch? And can this method be applied as an ODE solver where the forward and reverse processes are deterministic?

AlexiaJM commented 3 years ago

Sorry for the long delay. I added it, see the new ReadMe: https://github.com/AlexiaJM/score_sde_fast_sampling/blob/main/README.md#2021-09-21-using-the-gottagofast-sampler-in-torch

AlexiaJM commented 3 years ago

For ODE solving, you could use Heun's method instead of RK45, by changing the method in this line: https://github.com/yang-song/score_sde_pytorch/blob/1618ddea340f3e4a2ed7852a0694a809775cf8d0/sampling.py#L474. But ODE solving is much worse for VE than SDE solving , just keep that in mind.