(self, batch, **kwargs)
| 581 | return loss |
| 582 | |
| 583 | def shared_step(self, batch, **kwargs): |
| 584 | x, c, mask = self.get_input_withmask(batch, **kwargs) |
| 585 | # print(x.shape) |
| 586 | loss = self(x, c, mask=mask) |
| 587 | return loss |
| 588 | |
| 589 | @torch.no_grad() |
| 590 | def log_images(self, batch, N=8, n_row=4, sample=True, ddim_steps=200, ddim_eta=1., return_keys=None, |
no test coverage detected