exclude some more*2

dataset/shapenet.py

@@ -101,7 +101,7 @@ class CustomShapeNet(InMemoryDataset):
 
             for element in paths:
                 # This was build to filter all variations that aregreater then 25
-                pattern = re.compile('^((6[0-1]|[2-5][0-9])_\w+?\d+?|pc|\d+?_pc)\.(xyz|dat)$')
+                pattern = re.compile('^((6[0-1]|[1-5][0-9])_\w+?\d+?|pc|\d+?_pc)\.(xyz|dat)$')
                 if pattern.match(os.path.split(element)[-1]):
                     continue
                 else:

main.py

@@ -123,7 +123,8 @@ if __name__ == '__main__':
         print('Epoch {}, total epoches {}'.format(epoch+1, opt.nepoch))
 
         net.train()
-
+        # ToDo: We need different dataloader here to train the network in multiple iterations, maybe move the loop down
+        # for dataloader in ...
         for batch_idx, sample in enumerate(dataLoader):
             # points: (batch_size, n, 3)
             # labels: (batch_size, n)