def test_kpconvpretransform(self): params = load_model_config("segmentation", "kpconv", "SimpleKPConv") dataset = MockDatasetGeometric(5) model = instantiate_model(params, dataset) model.eval() dataset_transform = MockDatasetGeometric(5) dataset_transform.set_strategies(model) model.set_input(dataset[0]) model.forward() model.get_output() torch.testing.assert_allclose(dataset_transform[0].pos, dataset[0].pos)
def test_kpconvpretransform(self): params = load_model_config("segmentation", "kpconv")["PDSimpleKPConv"] model_config = OmegaConf.merge(params, self.data_config) dataset = MockDatasetGeometric(5) model_class = getattr(params, "class") model = _find_model_using_name(model_class, "segmentation", model_config, dataset) model.eval() dataset_transform = MockDatasetGeometric(5) dataset_transform.set_strategies(model) model.set_input(dataset[0]) model.forward() model.get_output() torch.testing.assert_allclose(dataset_transform[0].pos, dataset[0].pos)