(self, batch: dict, batch_idx: int)
| 293 | opt.step() |
| 294 | |
| 295 | def validation_step(self, batch: dict, batch_idx: int) -> Dict: |
| 296 | log_dict = self._validation_step(batch, batch_idx) |
| 297 | with self.ema_scope(): |
| 298 | log_dict_ema = self._validation_step(batch, batch_idx, postfix="_ema") |
| 299 | log_dict.update(log_dict_ema) |
| 300 | return log_dict |
| 301 | |
| 302 | def _validation_step(self, batch: dict, batch_idx: int, postfix: str = "") -> Dict: |
| 303 | x = self.get_input(batch) |
nothing calls this directly
no test coverage detected