(self, samples)
| 474 | self.model_ema(self.model) |
| 475 | |
| 476 | def _get_rows_from_list(self, samples): |
| 477 | n_imgs_per_row = len(samples) |
| 478 | denoise_grid = rearrange(samples, 'n b c h w -> b n c h w') |
| 479 | denoise_grid = rearrange(denoise_grid, 'b n c h w -> (b n) c h w') |
| 480 | denoise_grid = make_grid(denoise_grid, nrow=n_imgs_per_row) |
| 481 | return denoise_grid |
| 482 | |
| 483 | @torch.no_grad() |
| 484 | def log_images(self, batch, N=8, n_row=2, sample=True, return_keys=None, **kwargs): |