version 0.3.0

kentechx · Jul 10, 2023 · a2298f9 · a2298f9
1 parent 4fdc1c7
commit a2298f9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -39,7 +39,7 @@ Classification.
 import torch
 from x_dgcnn import DGCNN_Cls
 
-model = DGCNN_Cls(k=20, in_dim=3, out_dim=10)
+model = DGCNN_Cls(3, 10, 20)
 x = torch.randn(8, 3, 2048)
 xyz = x.clone()
 out = model(x, xyz)
@@ -52,7 +52,8 @@ Semantic segmentation.
 import torch
 from x_dgcnn import DGCNN_Seg
 
-model = DGCNN_Seg(k=40, in_dim=3, out_dim=10)
+# model = DGCNN_Seg(3, 10, 40, global_pooling=False)    # disable global pooling if batch size is too small
+model = DGCNN_Seg(3, 10, 40)
 x = torch.randn(8, 3, 2048)
 xyz = x.clone()
 out = model(x, xyz)

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
     name='x-dgcnn',
     packages=find_packages(),
-    version='0.2.2',
+    version='0.3.0',
     license='MIT',
     description='X-DGCNN - Pytorch',
     author='Kaidi Shen',