add multi-head point transformer layer

lucidrains · Jan 14, 2022 · 7881178 · 7881178
1 parent 16acc40
commit 7881178
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/point_transformer_pytorch/__init__.py b/point_transformer_pytorch/__init__.py
@@ -1 +1,2 @@
 from point_transformer_pytorch.point_transformer_pytorch import PointTransformerLayer
+from point_transformer_pytorch.multihead_point_transformer_pytorch import MultiheadPointTransformerLayer
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'point-transformer-pytorch',
   packages = find_packages(),
-  version = '0.0.3',
+  version = '0.1.0',
   license='MIT',
   description = 'Point Transformer - Pytorch',
   author = 'Phil Wang',
Original file line number	Diff line number	Diff line change
		@@ -1 +1,2 @@
		from point_transformer_pytorch.point_transformer_pytorch import PointTransformerLayer
		from point_transformer_pytorch.multihead_point_transformer_pytorch import MultiheadPointTransformerLayer