Sampler class for the FlowModel
(self, schedule, sample_eps=0)
| 224 | |
| 225 | class FlowSDE: |
| 226 | def __init__(self, schedule, sample_eps=0): |
| 227 | """ Sampler class for the FlowModel """ |
| 228 | self.sample_eps = sample_eps # velocity & [GVP, LINEAR] is stable everywhere, hence 0 |
| 229 | self.schedule = schedule |
| 230 | |
| 231 | def drift(self, x, t, model, **model_kwargs): |
| 232 | model_output = model(x, t, **model_kwargs) |
nothing calls this directly
no outgoing calls
no test coverage detected