(self, shape, device, noise_fn=torch.randn, constrain_fn=lambda x, t:x,
clip_denoised=False, max_timestep=None,
keep_running=False)
| 315 | return losses |
| 316 | |
| 317 | def gen_samples(self, shape, device, noise_fn=torch.randn, constrain_fn=lambda x, t:x, |
| 318 | clip_denoised=False, max_timestep=None, |
| 319 | keep_running=False): |
| 320 | return self.diffusion.p_sample_loop(self._denoise, shape=shape, device=device, noise_fn=noise_fn, |
| 321 | constrain_fn=constrain_fn, |
| 322 | clip_denoised=clip_denoised, max_timestep=max_timestep, |
| 323 | keep_running=keep_running) |
| 324 | |
| 325 | def reconstruct(self, x0, t, constrain_fn=lambda x, t:x): |
| 326 |
no test coverage detected