(self, batch: Any, batch_idx: int, dataloader_idx=0)
| 167 | self.mean_metrics.reset() |
| 168 | |
| 169 | def test_step(self, batch: Any, batch_idx: int, dataloader_idx=0): |
| 170 | task_name = self.task_list[dataloader_idx] |
| 171 | loss, outputs = self.step(task_name, batch) |
| 172 | |
| 173 | # log test metrics |
| 174 | self.test_metrics[task_name].update(outputs, **batch) |
| 175 | self.log( |
| 176 | f"test/{task_name}/loss", |
| 177 | loss, |
| 178 | on_step=False, |
| 179 | on_epoch=True, |
| 180 | add_dataloader_idx=False, |
| 181 | ) |
| 182 | |
| 183 | def test_epoch_end(self, outputs: List[Any]): |
| 184 | for metric in self.test_metrics.values(): |