(self, shape, device, noise_fn=torch.randn,
clip_denoised=True,
keep_running=False)
| 444 | return losses |
| 445 | |
| 446 | def gen_samples(self, shape, device, noise_fn=torch.randn, |
| 447 | clip_denoised=True, |
| 448 | keep_running=False): |
| 449 | return self.diffusion.p_sample_loop(self._denoise, shape=shape, device=device, noise_fn=noise_fn, |
| 450 | clip_denoised=clip_denoised, |
| 451 | keep_running=keep_running) |
| 452 | |
| 453 | def gen_sample_traj(self, shape, device, freq, noise_fn=torch.randn, |
| 454 | clip_denoised=True,keep_running=False): |