(self, x0, t, constrain_fn=lambda x, t:x)
| 323 | keep_running=keep_running) |
| 324 | |
| 325 | def reconstruct(self, x0, t, constrain_fn=lambda x, t:x): |
| 326 | |
| 327 | return self.diffusion.reconstruct(x0, t, self._denoise, constrain_fn=constrain_fn) |
| 328 | |
| 329 | def train(self): |
| 330 | self.model.train() |
nothing calls this directly
no outgoing calls
no test coverage detected