(metric_classes)
| 188 | |
| 189 | |
| 190 | def get_metric_values(metric_classes): |
| 191 | metric_values = {} |
| 192 | if is_master(): |
| 193 | for metric_cls in metric_classes: |
| 194 | metric_values.update(metric_cls.evaluate()) |
| 195 | if is_dist(): |
| 196 | metric_values = broadcast(metric_values, 0) |
| 197 | return metric_values |
| 198 | |
| 199 | |
| 200 | def collect_results_cpu(result_part, trainer, tmpdir=None): |