add permutator

lucidrains · Jun 24, 2021 · a36ecd6 · a36ecd6
1 parent d6901e7
commit a36ecd6
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -45,3 +45,14 @@ pred = model(img) # (1, 1000)
     primaryClass = {cs.CV}
 }
 ```
+
+```bibtex
+@misc{hou2021vision,
+    title   = {Vision Permutator: A Permutable MLP-Like Architecture for Visual Recognition},
+    author  = {Qibin Hou and Zihang Jiang and Li Yuan and Ming-Ming Cheng and Shuicheng Yan and Jiashi Feng},
+    year    = {2021},
+    eprint  = {2106.12368},
+    archivePrefix = {arXiv},
+    primaryClass = {cs.CV}
+}
+```
diff --git a/mlp_mixer_pytorch/__init__.py b/mlp_mixer_pytorch/__init__.py
@@ -1 +1,2 @@
 from mlp_mixer_pytorch.mlp_mixer_pytorch import MLPMixer
+from mlp_mixer_pytorch.permutator import Permutator
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'mlp-mixer-pytorch',
   packages = find_packages(exclude=[]),
-  version = '0.0.7',
+  version = '0.0.8',
   license='MIT',
   description = 'MLP Mixer - Pytorch',
   author = 'Phil Wang',
Original file line number	Diff line number	Diff line change
		@@ -1 +1,2 @@
		from mlp_mixer_pytorch.mlp_mixer_pytorch import MLPMixer
		from mlp_mixer_pytorch.permutator import Permutator