(self, batch, batch_idx)
| 267 | |
| 268 | @torch.no_grad() |
| 269 | def validation_step(self, batch, batch_idx): |
| 270 | lrm_generator_input = self.prepare_validation_batch_data(batch) |
| 271 | |
| 272 | render_out = self.forward(lrm_generator_input) |
| 273 | render_images = render_out['render_images'] |
| 274 | render_images = rearrange(render_images, 'b n c h w -> b c h (n w)') |
| 275 | |
| 276 | self.validation_step_outputs.append(render_images) |
| 277 | |
| 278 | def on_validation_epoch_end(self): |
| 279 | images = torch.cat(self.validation_step_outputs, dim=-1) |
nothing calls this directly
no test coverage detected