inference build

2020-06-23 21:05:49 +02:00
parent 1033b26195
commit 85cf3128f1
2 changed files with 4 additions and 7 deletions
--- a/datasets/shapenet.py
+++ b/datasets/shapenet.py
@ -198,10 +198,7 @@ class CustomShapeNet(InMemoryDataset):
                if self.collate_per_segment:
                    datasets[self.mode].append(data)
            if not self.collate_per_segment:
-                # This is just to be sure, but should not be needed, since src[all] == all
-                raise TypeError('FIX THIS')
-                # old Code
-                # datasets[self.mode].append(Data(**{key: torch.cat(data[key]) for key in data.keys()}))
+                datasets[self.mode].append(Data(**{key: torch.cat(data[key]) for key in data.keys()}))

        if datasets[self.mode]:
            os.makedirs(self.processed_dir, exist_ok=True)