pointnet2 working - TODO: Eval!

_templates/new_project/datasets/template_dataset.py

@@ -2,6 +2,11 @@ from torch.utils.data import Dataset
 
 
 class TemplateDataset(Dataset):
+
+    @property
+    def sample_shape(self):
+        return self[0][0].shape
+
     def __init__(self, *args, **kwargs):
         super(TemplateDataset, self).__init__()