(self, batch, batch_idx)
| 136 | return discloss |
| 137 | |
| 138 | def validation_step(self, batch, batch_idx): |
| 139 | log_dict = self._validation_step(batch, batch_idx) |
| 140 | with self.ema_scope(): |
| 141 | log_dict_ema = self._validation_step(batch, batch_idx, postfix="_ema") |
| 142 | return log_dict |
| 143 | |
| 144 | def _validation_step(self, batch, batch_idx, postfix=""): |
| 145 | inputs = self.get_input(batch, self.image_key) |
nothing calls this directly
no test coverage detected