Update README.md

congyingxia · Apr 8, 2019 · 830865e · 830865e
1 parent 2e12ab3
commit 830865e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -41,6 +41,12 @@ If you find our code useful, please cite our paper.
 }
 ```
 
+
+# Pytorch Version
+
+A pytorch version can be found here. https://github.com/nhhoang96/ZeroShotCapsule-PyTorch- 
+Thanks to Hoang Nguyen @nhhoang96.
+
 # Acknowledgements
 
 https://github.com/soskek/dynamic_routing_between_capsules