(self, x0, clip_denoised=True)
| 282 | return self.diffusion._prior_bpd(x0) |
| 283 | |
| 284 | def all_kl(self, x0, clip_denoised=True): |
| 285 | total_bpd_b, vals_bt, prior_bpd_b, mse_bt = self.diffusion.calc_bpd_loop(self._denoise, x0, clip_denoised) |
| 286 | |
| 287 | return { |
| 288 | 'total_bpd_b': total_bpd_b, |
| 289 | 'terms_bpd': vals_bt, |
| 290 | 'prior_bpd_b': prior_bpd_b, |
| 291 | 'mse_bt':mse_bt |
| 292 | } |
| 293 | |
| 294 | |
| 295 | def _denoise(self, data, t): |
nothing calls this directly
no test coverage detected