Method evaluate

diffplanner/datasets/base_dataset.py:109–115

(self, results, work_dir)

        self.eval_indexes = np.concatenate(self.eval_indexes)
            
    def evaluate(self, results, work_dir):
        metrics = {}
        device = results[0]['motion'].device
        for evaluator in self.evaluators:
            evaluator.to_device(device)
            metrics.update(evaluator.evaluate(results))
        return metrics