(self, batch, batch_idx)
| 170 | return discloss |
| 171 | |
| 172 | def validation_step(self, batch, batch_idx): |
| 173 | log_dict = self._validation_step(batch, batch_idx) |
| 174 | with self.ema_scope(): |
| 175 | log_dict_ema = self._validation_step(batch, batch_idx, suffix="_ema") |
| 176 | return log_dict |
| 177 | |
| 178 | def _validation_step(self, batch, batch_idx, suffix=""): |
| 179 | x = self.get_input(batch, self.image_key) |
nothing calls this directly
no test coverage detected