(trainer, data, metric_classes, vis_closure)
| 176 | |
| 177 | |
| 178 | def evaluate_batch(trainer, data, metric_classes, vis_closure): |
| 179 | batch_result = trainer.evaluation_step(data) |
| 180 | |
| 181 | if metric_classes is not None: |
| 182 | for metric_cls in metric_classes: |
| 183 | metric_cls.add(batch_result, data) |
| 184 | |
| 185 | if vis_closure is not None: |
| 186 | # trainer.visualization |
| 187 | vis_closure(batch_result) |
| 188 | |
| 189 | |
| 190 | def get_metric_values(metric_classes): |
no test coverage detected
searching dependent graphs…