(self)
| 1201 | return 'Final generating pseudo labels.' |
| 1202 | |
| 1203 | def run(self): |
| 1204 | if self.config.dataset.type == 'COCOStuffSegDatasetDev': |
| 1205 | results = self.test(self.model) |
| 1206 | results.update(self.test_with_TTA()) |
| 1207 | elif 'ParsingDataset' in self.config.dataset.type: |
| 1208 | evaluator = HumParEvaluator(dataset_name=self.ginfo.task_name, distributed=True, |
| 1209 | output_dir=self.ckpt_path, config=self.config) |
| 1210 | results = self.test(self.model, evaluator=evaluator) |
| 1211 | elif self.config.dataset.type == 'COCOPosDatasetDev': |
| 1212 | self.config.evaluation.cfg.name2id = self.dataset.name2id |
| 1213 | self.config.evaluation.cfg.dataset = self.dataset |
| 1214 | evaluator = PoseEvaluator(dataset_name=self.ginfo.task_name, distributed=True, |
| 1215 | output_dir=self.ckpt_path, config=self.config) |
| 1216 | results = self.test(self.model, evaluator=evaluator) |
| 1217 | else: |
| 1218 | raise NotImplementedError |
| 1219 | |
| 1220 | print(f"** results: {results}") |
| 1221 | |
| 1222 | default_collate_err_msg_format = ("default_collate: batch must contain tensors, numpy arrays, numbers, " |
| 1223 | "dicts or lists; found {}") |
nothing calls this directly
no test coverage detected