(self, outputs)
| 166 | return [self.wrong_tensor_name] |
| 167 | |
| 168 | def _on_fetches(self, outputs): |
| 169 | vec = outputs[0] |
| 170 | # TODO put shape assertion into inference-runner |
| 171 | assert vec.ndim == 1, "{} is not a vector!".format(self.wrong_tensor_name) |
| 172 | batch_size = len(vec) |
| 173 | wrong = np.sum(vec) |
| 174 | self.err_stat.feed(wrong, batch_size) |
| 175 | |
| 176 | def _after_inference(self): |
| 177 | return {self.summary_name: self.err_stat.ratio} |